forgot == 0

train.py

@@ -140,7 +140,7 @@ if __name__ == "__main__":
             batch_time = t1_batch - t0_batch
             batch_cumulated_time += batch_time
 
-            if batch_id % args.progress:
+            if batch_id % args.progress == 0:
                 print(
                     f"BATCH  {batch_id:05}/{epoch:04} - LOSS  {loss.item()} - TIME  {batch_cumulated_time}", flush=True)
                 batch_cumulated_time = timedelta()