debug 2.0

new_test_saved_finetuned_model.py

@@ -162,10 +162,8 @@ class BERTFineTuneTrainer:
                         logits = self.model.forward(data["input"], data["segment_label"], data["feat"])
 
                 logits = logits.cpu()
-                devic = logits.device  # or self.model.device if available
-                labels = data["label"].to(devic)
-
-                loss = self.criterion(logits, data["label"])
+                labels = data["label"].to(logits.device)
+                loss = self.criterion(logits, labels)
                 # if torch.cuda.device_count() > 1:
                 #     loss = loss.mean()