From 75881fd64d214280c9a5e41a49d016f9b1cf3d12 Mon Sep 17 00:00:00 2001
From: Logan Bauman <logan_bauman@brown.edu>
Date: Thu, 5 May 2022 00:11:00 -0400
Subject: mega

---
 main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'main.py')

diff --git a/main.py b/main.py
index de769baa..60574e63 100644
--- a/main.py
+++ b/main.py
@@ -36,7 +36,8 @@ def parse_args():
     return parser.parse_args()
 
 def train(model):
-    for _ in range(hp.num_epochs):
+    for i in range(hp.num_epochs):
+        print('batch', i)
         model.train_step()
 
 def main():
-- 
cgit v1.2.3-70-g09d2