Fix input shape of generator
Use self.latent_dim for generator's input shape rather than a constant 100.
This commit is contained in:
Родитель
2ccd7cd4ec
Коммит
b20a4b8461
|
@ -35,7 +35,7 @@ class DCGAN():
|
|||
self.generator = self.build_generator()
|
||||
|
||||
# The generator takes noise as input and generates 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
|
||||
|
|
Загрузка…
Ссылка в новой задаче