updates

2019-05-30 20:21:25 +02:00 · 2019-05-30 20:21:25 +02:00 · ea11af6132
commit ea11af6132
parent 70ba7f0805
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -287,7 +287,7 @@ def train(
            del chkpt

    dt = (time.time() - t0) / 3600
-    print('%g epochs completed in %.3f hours.' % (epoch - start_epoch, dt))
+    print('%g epochs completed in %.3f hours.' % (epoch - start_epoch + 1, dt))
    return results