From 2b92b90ed6f5e8a8d8760eee4c4a253adc9a07b1 Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 15:06:37 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index e2f8f0b..e8da996 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -94,8 +94,8 @@ with tf.device('/gpu:0'): def call(self, inputs, embed=True): if embed: dtype = tf.keras.backend.dtype(inputs) - if dtype != 'int32' and dtype != 'int64': - inputs = math_ops.cast(inputs, 'int32') + # if dtype != 'int32' and dtype != 'int64': + # inputs = math_ops.cast(inputs, 'int32') return embedding_ops.embedding_lookup(self.w, inputs) else: return tf.tensordot(inputs, tf.transpose(self.w), 1) + self.b