From 541ea1d7b1d168fe7b4588d57c6691241f9d9f4f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= Date: Fri, 1 Apr 2022 12:11:37 +1300 Subject: [PATCH] Add missing argument n_epochs=50 --- 17_autoencoders_gans_and_diffusion_models.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/17_autoencoders_gans_and_diffusion_models.ipynb b/17_autoencoders_gans_and_diffusion_models.ipynb index 28188ac..b3fb4cb 100644 --- a/17_autoencoders_gans_and_diffusion_models.ipynb +++ b/17_autoencoders_gans_and_diffusion_models.ipynb @@ -2262,7 +2262,7 @@ "dataset = tf.data.Dataset.from_tensor_slices(X_train_dcgan)\n", "dataset = dataset.shuffle(1000)\n", "dataset = dataset.batch(batch_size, drop_remainder=True).prefetch(1)\n", - "train_gan(gan, dataset, batch_size, codings_size)" + "train_gan(gan, dataset, batch_size, codings_size, n_epochs=50)" ] }, {