From 16c5a20f28bafed73c5d779cfb99f0a6fdb975a0 Mon Sep 17 00:00:00 2001
From: Ero98 <oujghg@163.com>
Date: Thu, 2 Aug 2018 17:47:13 +0800
Subject: [PATCH] Update lsgan.py

---
 lsgan/lsgan.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lsgan/lsgan.py b/lsgan/lsgan.py
index 6014f51..09e9683 100644
--- a/lsgan/lsgan.py
+++ b/lsgan/lsgan.py
@@ -34,7 +34,7 @@ class LSGAN():
         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