From b076a0d12311911355cdabbb9e23b852db5016c8 Mon Sep 17 00:00:00 2001
From: k <markers711@gmail.com>
Date: Wed, 12 Nov 2025 12:12:26 -0500
Subject: [PATCH] add status bar for epoch progress

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

diff --git a/train.py b/train.py
index e912da5..43f91e7 100644
--- a/train.py
+++ b/train.py
@@ -6,6 +6,7 @@ import matplotlib.pyplot as plt
 import time
 import show
 from model import gen
+from tqdm import tqdm
 
 BATCH_SIZE = 16
 EPOCHS = 100
@@ -48,7 +49,7 @@ eshape = (BATCH_SIZE, 1, 128, 216)
 for epoch in range(0,EPOCHS):
     print(f"\n--- Starting Epoch {epoch} ---\n")
     loss=0
-    for i in range(0,len(x),BATCH_SIZE):
+    for i in tqdm(range(0,len(x),BATCH_SIZE)):
         tx=Tensor(x[i:i+BATCH_SIZE])
         ty=Tensor(y[i:i+BATCH_SIZE])
         if(tx.shape != eshape):