diff --git a/train.py b/train.py
index ce3527a..5069f9b 100644
--- a/train.py
+++ b/train.py
@@ -230,7 +230,7 @@ def main():
         # Finished an epoch, evaluate mAP and mAR
         if(progress_epoch >= len(train_set)):
             epochs_trained += 1
-            progress_epoch = 0
+            progress_epoch -= len(train_set)
 
             print("----- Finished epoch -----")
             print("Num epochs trained:", epochs_trained)