jessicayjm
commited on
Commit
•
a74b362
1
Parent(s):
6334b7a
update sample code
Browse files
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')
|