debug 3.0

new_test_saved_finetuned_model.py

@@ -183,7 +183,8 @@ class BERTFineTuneTrainer:
                 tlabels.extend(data['label'].cpu().numpy())
                 positive_class_probs = [prob[1] for prob in probabs]
                 # Compare predicted labels to true labels and calculate accuracy
-                correct = (data['label'] == predicted_labels).sum().item()
+                correct = (data['label'].to(predicted_labels.device) == predicted_labels).sum().item()
+
                 
                 avg_loss += loss.item()
                 total_correct += correct