snapshot...

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())