This commit is contained in:
nickwalton
2019-10-22 15:10:38 -06:00
parent 2b92b90ed6
commit a0f200ec5a
+1 -1
View File
@@ -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()