Skip to content
Snippets Groups Projects
Commit 717cad69 authored by tomrink's avatar tomrink
Browse files

minor

parent 8ae90c64
No related branches found
Tags gamma-test-r20220119_2201
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment