From a366d28adeaf38aede24c657c9ef4becaa10d952 Mon Sep 17 00:00:00 2001 From: jh Date: Tue, 16 Mar 2021 16:36:52 -0700 Subject: [PATCH] debug shape --- models/Model_SAEHD/Model.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 25d6228..b47e339 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -495,6 +495,9 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... gpu_pred_src_src_d, \ gpu_pred_src_src_d2 = self.D_src(gpu_pred_src_src_masked_opt) + print("Shape (gpu_pred_src_src_d): ", gpu_pred_src_src_d.shape) + print("Shape (gpu_pred_src_src_d2): ", gpu_pred_src_src_d2.shape) + gpu_pred_src_src_d_ones = tf.ones_like (gpu_pred_src_src_d) gpu_pred_src_src_d_zeros = tf.zeros_like(gpu_pred_src_src_d)