diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 508c88ecf9a6d9f4d348e30d38d1ee77fdd9593c..3cffe4751abb1148bc2f9355cfa7cbfd6266f955 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -905,7 +905,7 @@ class IcingIntensityFCN: with self.writer_train.as_default(): tf.summary.scalar('loss_trn', loss.numpy(), step=step) - tf.summary.scalar('learning_rate', self.optimizer._decayed_lr('float32').numpy(), step=step) + tf.summary.scalar('learning_rate', self.optimizer.lr.numpy(), step=step) tf.summary.scalar('num_train_steps', step, step=step) tf.summary.scalar('num_epochs', epoch, step=step) @@ -936,7 +936,7 @@ class IcingIntensityFCN: tf.summary.scalar('loss_val', self.test_loss.result(), step=step) print('****** test loss, acc, lr: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy(), - self.optimizer._decayed_lr('float32').numpy()) + self.optimizer.lr.numpy()) step += 1 print('train loss: ', loss.numpy())