From 717cad6950607933cf24c0bca03a65f673a2b439 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 5 Nov 2021 13:27:25 -0500
Subject: [PATCH] minor
---
modules/deeplearning/icing_cnn.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 2d0d8de8..f0e57e2f 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -805,6 +805,7 @@ class IcingIntensityNN:
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('num_train_steps', step, step=step)
tf.summary.scalar('num_epochs', epoch, step=step)
--
GitLab