diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index eb58b5bc27541679683e814428e25608fda05eac..199f8ab9fbb4a1743f932dc711d9d5949bbf02fe 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -46,6 +46,7 @@ DO_ZERO_OUT = False CACHE_FILE = '' USE_EMA = True EMA_OVERWRITE_FREQUENCY = 5 +EMA_MOMENTUM = 0.99 # setup scaling parameters dictionary mean_std_dct = {} @@ -498,6 +499,7 @@ class SRCNN: optimizer = tf.keras.optimizers.Adam(learning_rate=self.learningRateSchedule, use_ema=USE_EMA, + ema_momentum=EMA_MOMENTUM, ema_overwrite_frequency=EMA_OVERWRITE_FREQUENCY) self.optimizer = optimizer