diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index 0aca1b76c9661ec9d8b15710a6dab59334e6ebed..bdc94fca87d5a3ac518df45368d4b5794280c158 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -609,7 +609,7 @@ class SRCNN: 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) @@ -629,7 +629,7 @@ class SRCNN: 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())