From 6092a77924f34762b8c4769b753d5a4136862664 Mon Sep 17 00:00:00 2001 From: jh Date: Tue, 16 Mar 2021 16:57:53 -0700 Subject: [PATCH] more debugging --- models/Model_SAEHD/Model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 2bfed55..5d48762 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -497,8 +497,9 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... 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) + for i in gpu_pred_src_src_d2.shape: + print("Shape (gpu_pred_src_src_d2) i: ", i) - print("Flattened Shape (gpu_pred_src_src_d): ", gpu_pred_src_src_d.flatten().shape()) # def get_smooth_noisy_labels_like(label, tensor, smoothing=0.2, noise=0.05): # new_labels = []