sohomghosh
/

LIPI_FinSim4_ESG_task2

Model card Files Files and versions

sohomghosh commited on Jun 27, 2022

Commit

8a57cfb

·

1 Parent(s): 7ffaed7

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -117,7 +117,8 @@ def do_predict(tokenizer):
   return actual_predictions
 model_sus = BERTClass(2)
-model_sus.load_stat_dict(torch.load('pytorch_model.bin')['model_state_dict'])
 tokenizer_sus = BertTokenizer.from_pretrained('roberta-base')
 actual_predictions_sus = do_predict(tokenizer_sus)

   return actual_predictions
 model_sus = BERTClass(2)
+model_sustain.to(device)
+model_sustain.load_state_dict(torch.load('pytorch_model.bin', map_location=device)['model_state_dict'])
 tokenizer_sus = BertTokenizer.from_pretrained('roberta-base')
 actual_predictions_sus = do_predict(tokenizer_sus)