dbg

new_test_saved_finetuned_model.py

@@ -162,6 +162,9 @@ 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"])
                 # if torch.cuda.device_count() > 1:
                 #     loss = loss.mean()