diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py index b0b48de5d6d6dc2c57b846a8d93dec44eb3de9c4..d91d30c69932a0a10d0e5354959fad599e0ccc8e 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py @@ -467,7 +467,7 @@ class SRCNN: self.loss = tf.keras.losses.MeanSquaredError() # Regression # decayed_learning_rate = learning_rate * decay_rate ^ (global_step / decay_steps) - initial_learning_rate = 0.002 + initial_learning_rate = 0.001 decay_rate = 0.95 steps_per_epoch = int(self.num_data_samples/BATCH_SIZE) # one epoch decay_steps = int(steps_per_epoch) * 4