From a89cf63e9f7adc1c380fc473ebaa6d9ced5a6a9c Mon Sep 17 00:00:00 2001 From: jh Date: Tue, 16 Mar 2021 19:20:24 -0700 Subject: [PATCH] fix reshape --- 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 553e6e8..7cd65a6 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -514,7 +514,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... else: label = np.random.uniform(1-smoothing/2, 1.0) labels.append(label) - return tf.reshape(labels, tensor.shape) + return tf.reshape(labels, (self.batch_size,) + tensor.shape[1:]) gpu_pred_src_src_d_ones = get_smooth_noisy_labels(1, gpu_pred_src_src_d, smoothing=0.2, noise=0.05) gpu_pred_src_src_d_zeros = get_smooth_noisy_labels(0, gpu_pred_src_src_d, smoothing=0.2, noise=0.05)