Commit 933846d3 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

forgot == 0

parent 5e3bb780
......@@ -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()
......
Supports Markdown
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