diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index e3e628dfe244470e2eddbc6083bba00931e43856..eb58b5bc27541679683e814428e25608fda05eac 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -42,7 +42,8 @@ DO_AUGMENT = True DO_SMOOTH = False SIGMA = 1.0 DO_ZERO_OUT = False -CACHE_FILE = '/scratch/long/rink/cld_opd_abi_128x128_cache' +# CACHE_FILE = '/scratch/long/rink/cld_opd_abi_128x128_cache' +CACHE_FILE = '' USE_EMA = True EMA_OVERWRITE_FREQUENCY = 5 @@ -483,8 +484,8 @@ class SRCNN: def build_training(self): - self.loss = tf.keras.losses.MeanSquaredError() # Regression - # self.loss = tf.keras.losses.MeanAbsoluteError() # Regression + # self.loss = tf.keras.losses.MeanSquaredError() # Regression + self.loss = tf.keras.losses.MeanAbsoluteError() # Regression # decayed_learning_rate = learning_rate * decay_rate ^ (global_step / decay_steps) initial_learning_rate = 0.001