1 files changed, 1 insertions, 0 deletions
diff --git a/cgan.py b/cgan.py
index e0c1e4e..d44f961 100644
--- a/cgan.py
+++ b/cgan.py
@@ -207,6 +207,7 @@ class CGAN():
       train_data = np.pad(train_data, ((0,0),(2,2),(2,2),(0,0)), 'constant')
       labels_test = labels_test.flatten()
       labels_train = labels_train.flatten()
+      print(train_data.shape, test_data.shape, labels_test.shape, labes_train.shape)
       return train_data, test_data, labels_train, labels_test