snapshot...

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)