From a0f200ec5ab7942e2c173691d29e9a1523072be4 Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 15:10:38 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index e8da996..dd94e8c 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -102,7 +102,7 @@ with tf.device('/gpu:0'): # input for the keras model - tokens = tf.keras.layers.Input(shape=(seq_length,), dtype='int32') + tokens = tf.keras.layers.Input(shape=(seq_length,), dtype='float32') tied_embedding_softmax = TiedEmbeddingSoftmax()