Logs

app.py

@@ -123,6 +123,11 @@ def evaluate_model(model, dataloader, device):
 
     # Disable gradient calculations
     with torch.no_grad():
+        for batch in dataloader:
+            print("Sample sequence labels:", batch['labels'][0].tolist()[:20])
+            print("Corresponding predictions:", torch.argmax(model(batch['input_ids'].to(device), 
+                attention_mask=batch['attention_mask'].to(device)).logits, dim=-1)[0].tolist()[:20])
+            break
         for batch in dataloader:
             input_ids = batch['input_ids'].to(device)
             attention_mask = batch['attention_mask'].to(device)