1 files changed, 1 insertions, 1 deletions
diff --git a/cgan.py b/cgan.py
index 17aa367..eb0dbc3 100644
--- a/cgan.py
+++ b/cgan.py
@@ -208,7 +208,7 @@ class CGAN():
 
       train_data = self.generator.predict([noise_train, labels_train])
       test_data = self.generator.predict([noise_test, labels_test])
-      val_data = self.generator.predict([noise_train, labels_val])
+      val_data = self.generator.predict([noise_val, labels_val])
       return train_data, test_data, val_data, labels_train, labels_test, labels_val