Logs

app.py

@@ -122,6 +122,7 @@ def evaluate_model(model, dataloader, device):
     all_labels = []
 
     test = True
+    test2 = True
     # Disable gradient calculations
     with torch.no_grad():
         for batch in dataloader:
@@ -135,8 +136,12 @@ def evaluate_model(model, dataloader, device):
 
             # Get predictions
             preds = torch.argmax(logits, dim=-1).cpu().numpy()
-            if test:
+            if test or test2:
+                if test:
+                    test2 = False 
                 test = False
+                print(len(preds))
+                print(len(labels))
                 print(preds)
                 print(labels)