Added epoch tracking in terminal

Johanmkr · Johanmkr · commit af3d8bc91819 · 2025-02-20T13:56:49.000+01:00
diff --git a/main.py b/main.py
@@ -139,6 +139,7 @@ def main():
 
     for epoch in range(args.epoch):
         # Training loop start
+        print(f"Epoch: {epoch+1}/{args.epoch}")
         trainingloss = []
         model.train()
         for x, y in tqdm(trainloader, desc="Training"):
