diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 8d8d009ca90ef15381999478dc34b78beaf3b302..89166211d56adf2e22b520bfaa9e1214c1c14fd2 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/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()