From 7eec33f756cffee0c4526c7e3e6af71246dd0787 Mon Sep 17 00:00:00 2001 From: nunzip Date: Wed, 27 Feb 2019 18:48:24 +0000 Subject: debug cgan --- cgan.py | 1 + 1 file changed, 1 insertion(+) 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 -- cgit v1.2.3