Add missing argument n_epochs=50

2022-04-01 12:11:37 +13:00 · 2022-04-01 12:11:37 +13:00 · 541ea1d7b1
parent 05e891c997
commit 541ea1d7b1
1 changed files with 1 additions and 1 deletions
--- 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)"
   ]
  },
  {