diff --git a/train.py b/train.py
index 5aaa48b1..e01f4c00 100644
--- 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