diff --git a/train.py b/train.py index 501045e..81d4d00 100644 --- a/train.py +++ b/train.py @@ -222,7 +222,7 @@ for epoch in range(num_epochs): time_left = datetime.timedelta(seconds=batches_left * (time.time() - prev_time)) epoch_time = time.time() - prev_time prev_time = time.time() - if i % 100 == 0: + if i % 1 == 0: sys.stdout.write( "\r[Epoch %d/%d] [Batch %d/%d] [loss: %f] ETA: %.10s" % (