diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 9bb4deab887f3f052c3b34f0b0463b48eea94ba1..613eadf3a11180b69ef653274968c44ad22f9293 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -394,7 +394,7 @@ class IcingIntensityNN: 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 = 4 * steps_per_epoch + decay_steps = 8 * steps_per_epoch 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)