diff --git a/modules/deeplearning/cloud_opd_srcnn.py b/modules/deeplearning/cloud_opd_srcnn.py index 17f2e5960abfe61550cbba30887bf5b9aba0a54b..f2f35275914ec1f6574439d4e1f56750001c87f6 100644 --- a/modules/deeplearning/cloud_opd_srcnn.py +++ b/modules/deeplearning/cloud_opd_srcnn.py @@ -258,7 +258,7 @@ class SRCNN: self.test_label_files = None # self.n_chans = len(data_params_half) + len(data_params_full) + 1 - self.n_chans = 5 + self.n_chans = 3 self.X_img = tf.keras.Input(shape=(None, None, self.n_chans)) @@ -467,7 +467,7 @@ class SRCNN: 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 + decay_steps = int(steps_per_epoch) * 4 print('initial rate, decay rate, steps/epoch, decay steps: ', initial_learning_rate, decay_rate, steps_per_epoch, decay_steps) self.learningRateSchedule = tf.keras.optimizers.schedules.ExponentialDecay(initial_learning_rate, decay_steps, decay_rate)