aboutsummaryrefslogtreecommitdiff
path: root/cgan.py
diff options
context:
space:
mode:
Diffstat (limited to 'cgan.py')
-rw-r--r--cgan.py1
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