Log the epoch results

jshuadvd · Jul 6, 2024 · 7b00be2 · 7b00be2
1 parent ee2e5f3
commit 7b00be2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -248,8 +248,8 @@ def train(
             }
         )
 
-        # Print epoch results
-        print(
+        # Log epoch results
+        logger.info(
             f"Epoch {epoch+1}, Train Loss: {avg_train_loss:.4f}, Train Perplexity: {train_perplexity:.4f}, "
             f"Val Loss: {avg_val_loss:.4f}, Val Perplexity: {val_perplexity:.4f}"
         )