jessicayjm commited on
Commit
a74b362
1 Parent(s): 6334b7a

update sample code

Browse files
Files changed (1) hide show
  1. README.md +0 -2
README.md CHANGED
@@ -51,8 +51,6 @@ from openprompt import PromptForClassification
51
  from openprompt.data_utils import InputExample
52
  from openprompt import PromptDataLoader
53
 
54
-
55
- torch.cuda.set_device(1)
56
  checkpoint_file = 'upload_version/empathy-appraisal-span.pt'
57
 
58
  plm, tokenizer, model_config, WrapperClass = load_plm('roberta', 'roberta-large')
 
51
  from openprompt.data_utils import InputExample
52
  from openprompt import PromptDataLoader
53
 
 
 
54
  checkpoint_file = 'upload_version/empathy-appraisal-span.pt'
55
 
56
  plm, tokenizer, model_config, WrapperClass = load_plm('roberta', 'roberta-large')