snapshot...

modules/deeplearning/icing_fcn.py

@@ -26,6 +26,8 @@ NUM_EPOCHS = 60
 
 
 EARLY_STOP = True
+PATIENCE = 7
+
 USE_EMA = False
 EMA_OVERWRITE_FREQUENCY = 5
 EMA_MOMENTUM = 0.99
@@ -883,7 +885,7 @@ class IcingIntensityFCN:
         best_test_mcc = 0
 
         if EARLY_STOP:
-            es = EarlyStop()
+            es = EarlyStop(patience=PATIENCE)
 
         for epoch in range(NUM_EPOCHS):
             self.train_loss.reset_states()