From 6ac6d368bdc154859b6f9aafd572894450ce2a4c Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 19 Oct 2023 11:08:29 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 508c88ec..3cffe475 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())
--
GitLab