Fix for epoch tracking

gwinndr · Aug 23, 2020 · b4f4828 · b4f4828
1 parent 4f91d34
commit b4f4828
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)