Update lsgan.py

fabclmnt · Aug 2, 2018 · 16c5a20 · 16c5a20
1 parent 0106928
commit 16c5a20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lsgan/lsgan.py b/lsgan/lsgan.py
@@ -34,7 +34,7 @@ def __init__(self):
         self.generator = self.build_generator()
 
         # The generator takes noise as input and generated imgs
-        z = Input(shape=(100,))
+        z = Input(shape=(self.latent_dim,))
         img = self.generator(z)
 
         # For the combined model we will only train the generator