fix

GreenWizard2015 · Jun 4, 2024 · aa0b509 · aa0b509
1 parent ed74a15
commit aa0b509
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/train.py b/scripts/train.py
@@ -293,6 +293,7 @@ def performRandomSearch(epoch=0):
         break
       if 'reset' == args.on_patience:
         print('Resetting the model to the average of the best models')
+        bestEpoch = epoch # reset the patience
         performRandomSearch(epoch=epoch)
     continue
   return