From 94a53c69c25a511048b1ea84d832b29fc7dcc5e2 Mon Sep 17 00:00:00 2001 From: jh Date: Sat, 8 May 2021 12:40:00 -0700 Subject: [PATCH] fix --- models/Model_SAEHD/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 3907a02..e872ae1 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -580,7 +580,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... x = tf.cast(x, tf.float32) x = tf.math.scalar_mul(1-smoothing, x) # x = x + (smoothing/num_labels) - x = tf.reshape(x, (self.batch_size,) + tensor.get_shape().as_list()[1:]) + x = tf.reshape(x, (self.batch_size,) + tuple(tensor.get_shape().as_list()[1:])) return x smoothing = self.options['gan_smoothing']