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')
|