Update code/do_predict.py

code/do_predict.py

@@ -20,7 +20,7 @@ model_name = "roberta_softmax.pt"
 device = torch.device("cuda", 0) if torch.cuda.is_available() else torch.device("cpu")
 model_roberta = Model_Softmax(config).to(device)
 model_roberta.load_state_dict(
-    state_dict=torch.load(directory + model_name, map_location=device)
+    state_dict=torch.load(directory + model_name, map_location=device), strict=False
 )