From c8d8ec99289c1c9c6c546a84600eb830fae1fcd3 Mon Sep 17 00:00:00 2001
From: Kenneth <kennethnym@outlook.com>
Date: Sun, 19 May 2024 00:07:34 +0100
Subject: [PATCH] fix batch loss reporting

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index bd775a5..d1caa7d 100644
--- a/train.py
+++ b/train.py
@@ -173,8 +173,8 @@ def train():
             accuracy = 100 * correct / sample_size
 
             running_loss += loss.item()
-            if i % 1000 == 999:
-                last_loss = running_loss / 1000  # loss per batch
+            if i % 10 == 9:
+                last_loss = running_loss / 10  # loss per batch
                 print("  batch {} loss: {}".format(i + 1, last_loss))
                 running_loss = 0.0