diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py index 63dfec78068e05bca2dcdcc62a19705ea6494f8a..efe15ca3c67901b6a3c966e391b65c815a09d10f 100644 --- a/modules/deeplearning/unet.py +++ b/modules/deeplearning/unet.py @@ -748,7 +748,7 @@ class UNET: self.loss = tf.keras.losses.MeanSquaredError() # Regression # decayed_learning_rate = learning_rate * decay_rate ^ (global_step / decay_steps) - initial_learning_rate = 0.006 + initial_learning_rate = 0.002 decay_rate = 0.95 steps_per_epoch = int(self.num_data_samples/BATCH_SIZE) # one epoch decay_steps = int(steps_per_epoch / 2)