diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 32465e67bca90a7aa498cf647b62ed5dd45180db..5990b0a49f900320177cefda32756552832289e2 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -656,6 +656,9 @@ class IcingIntensityNN: self.optimizer = optimizer self.initial_learning_rate = initial_learning_rate + print('initial_learning_rate, decay_rate: ', initial_learning_rate, decay_rate) + print('steps_per_epoch, decay_steps: ', steps_per_epoch, decay_steps) + def build_evaluation(self): self.train_loss = tf.keras.metrics.Mean(name='train_loss') self.test_loss = tf.keras.metrics.Mean(name='test_loss')