Commit 6d494e7f authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

Display len(train_loader)

parent 933846d3
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment