diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index a014d4174695134b6183d586c1cb7d327fd83f87..59d37ec432042e980858dbc63ccdd95655f4e9c9 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -830,7 +830,8 @@ class IcingIntensityNN: tf.summary.scalar('num_train_steps', step, step=step) tf.summary.scalar('num_epochs', epoch, step=step) - print('****** test loss, acc: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy()) + print('****** test loss, acc, lr: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy(), + self.optimizer._decayed_lr('float32').numpy()) step += 1 print('train loss: ', loss.numpy())