Skip to content
Snippets Groups Projects
Commit 30323274 authored by Jean-Luc Parouty's avatar Jean-Luc Parouty
Browse files

Update GAN

parent bc99ce0c
No related branches found
No related tags found
No related merge requests found
...@@ -401,7 +401,7 @@ ...@@ -401,7 +401,7 @@
"name": "python", "name": "python",
"nbconvert_exporter": "python", "nbconvert_exporter": "python",
"pygments_lexer": "ipython3", "pygments_lexer": "ipython3",
"version": "3.9.7" "version": "3.9.2"
} }
}, },
"nbformat": 4, "nbformat": 4,
......
...@@ -63,6 +63,13 @@ class DCGAN(keras.Model): ...@@ -63,6 +63,13 @@ class DCGAN(keras.Model):
discriminator_optimizer = keras.optimizers.Adam(), discriminator_optimizer = keras.optimizers.Adam(),
generator_optimizer = keras.optimizers.Adam(), generator_optimizer = keras.optimizers.Adam(),
loss_function = keras.losses.BinaryCrossentropy() ): loss_function = keras.losses.BinaryCrossentropy() ):
'''
Compile the model
args:
discriminator_optimizer : Discriminator optimizer (Adam)
generator_optimizer : Generator optimizer (Adam)
loss_function : Loss function
'''
super(DCGAN, self).compile() super(DCGAN, self).compile()
self.d_optimizer = discriminator_optimizer self.d_optimizer = discriminator_optimizer
self.g_optimizer = generator_optimizer self.g_optimizer = generator_optimizer
...@@ -147,7 +154,7 @@ class DCGAN(keras.Model): ...@@ -147,7 +154,7 @@ class DCGAN(keras.Model):
# Sample random points in the latent space # Sample random points in the latent space
random_latent_vectors = tf.random.normal(shape=(batch_size, self.latent_dim)) random_latent_vectors = tf.random.normal(shape=(batch_size, self.latent_dim))
# Assemble labels that say "all real images" # Assemble labels that say all images are real, yes it's a lie ;-)
misleading_labels = tf.zeros((batch_size, 1)) misleading_labels = tf.zeros((batch_size, 1))
# ---- Train the generator --------------------------------- # ---- Train the generator ---------------------------------
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment