fix

GreenWizard2015 · GreenWizard2015 · commit aa0b509bc95f · 2024-06-04T18:09:23.000+02:00
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
