Skip to content

Commit

Permalink
change input dim in critic to use latent_dim variable
Browse files Browse the repository at this point in the history
  • Loading branch information
Andy Wu committed Jul 23, 2019
1 parent 44d3320 commit e0bae5b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion wgan_gp/wgan_gp.py
Expand Up @@ -93,7 +93,7 @@ def __init__(self):
self.generator.trainable = True

# Sampled noise for input to generator
z_gen = Input(shape=(100,))
z_gen = Input(shape=(self.latent_dim,))
# Generate images based of noise
img = self.generator(z_gen)
# Discriminator determines validity
Expand Down

0 comments on commit e0bae5b

Please sign in to comment.