From 7bad170add38c9b5ba49bc2e7297dc8072b28dd3 Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 18:48:12 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index e723040..734d4e3 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -133,7 +133,9 @@ model = tf.keras.Model(inputs=tokens, outputs=logits) # the loss function is a simple categorical crossentropy between the logits and the labels def loss(labels, logits): - return tf.keras.losses.sparse_categorical_crossentropy(labels, logits, from_logits=True) + loss = tf.keras.losses.sparse_categorical_crossentropy(labels, logits, from_logits=True) + loss = tf.Print(loss, [loss]) + return loss # the optimizer is not used since this code only supports inference