wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-09 11:13:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
nickwalton
2019-10-22 18:48:12 -06:00
parent cba5bd1827
commit 7bad170add
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/ctrl/training_utils/training.py
+3 -1
View File
@@ -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