frugalai-submission-template

Runtime error

0xrushi commited on Jan 14

Commit

f40d4de

verified ·

1 Parent(s): 172535d

Update tasks/text.py

Files changed (1) hide show

tasks/text.py CHANGED Viewed

@@ -48,7 +48,7 @@ model = RobertaForSequenceClassification.from_pretrained(
     "roberta-base",
     num_labels=8
 )
-model.load_state_dict(torch.load('task/best_roberta_model.pth', map_location=device))
 model.to(device)
 model.eval()

     "roberta-base",
     num_labels=8
 )
+model.load_state_dict(torch.load('tasks/best_roberta_model.pth', map_location=device))
 model.to(device)
 model.eval()