Update cgan.py

kavehahmadi60 · Aug 2, 2018 · ebbd008 · ebbd008
1 parent bdd3c40
commit ebbd008
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cgan/cgan.py b/cgan/cgan.py
@@ -35,7 +35,7 @@ def __init__(self):
 
         # The generator takes noise and the target label as input
         # and generates the corresponding digit of that label
-        noise = Input(shape=(100,))
+        noise = Input(shape=(self.latent_dim,))
         label = Input(shape=(1,))
         img = self.generator([noise, label])