From ebbd0084d7f08e39ea7b5c0f61a50d2b8511fa7e Mon Sep 17 00:00:00 2001 From: Ero98 Date: Thu, 2 Aug 2018 17:44:46 +0800 Subject: [PATCH] Update cgan.py --- cgan/cgan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cgan/cgan.py b/cgan/cgan.py index 0cc9924d47..e66acb6e4a 100755 --- 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])