From c2bdf2652088e5a648a5ab574bf760ef05ba69c6 Mon Sep 17 00:00:00 2001 From: jh Date: Sat, 8 May 2021 12:37:27 -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 33eaf1d..3907a02 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.shape.as_list()[1:]) + x = tf.reshape(x, (self.batch_size,) + tensor.get_shape().as_list()[1:]) return x smoothing = self.options['gan_smoothing']