Display len(train_loader)

train.py

@@ -142,7 +142,7 @@ if __name__ == "__main__":
 
             if batch_id % args.progress == 0:
                 print(
-                    f"BATCH  {batch_id:05}/{epoch:04} - LOSS  {loss.item()} - TIME  {batch_cumulated_time}", flush=True)
+                    f"BATCH  {batch_id:05}/{len(train_loader):05} - LOSS  {loss.item()} - TIME  {batch_cumulated_time}", flush=True)
                 batch_cumulated_time = timedelta()
 
         writer.add_scalar("Loss/train", cumloss / len(train_loader), epoch)