printing

kyegomez · Oct 2, 2023 · dd2178a · dd2178a
1 parent 71213c8
commit dd2178a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -96,7 +96,10 @@ def validate_epoch(model, loader, criterion, device):
 
 for epoch in range(num_epochs):
     train_loss, train_acc = train_epoch(model, train_loader, optimizer, criterion, device)
+    print(train_loss, train_acc)
+
     val_loss, val_acc = validate_epoch(model, val_loader, criterion, device)
+    print(val_loss, val_acc)
 
     print(f"Epoch: {epoch+1}/{num_epochs}, Train Loss: {train_loss:.4f}, Train Acc: {train_acc:.4f}, Val Loss: {val_loss:.4f}, Val Acc: {val_acc:.4f}")