snapshot...

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