schuetoliver
/
handson-ml
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add missing argument n_epochs=50

main
Aurélien Geron 2022-04-01 12:11:37 +13:00
parent 05e891c997
commit 541ea1d7b1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
17_autoencoders_gans_and_diffusion_models.ipynb
View File

@ -2262,7 +2262,7 @@
"dataset = tf.data.Dataset.from_tensor_slices(X_train_dcgan)\n", "dataset = tf.data.Dataset.from_tensor_slices(X_train_dcgan)\n",
"dataset = dataset.shuffle(1000)\n", "dataset = dataset.shuffle(1000)\n",
"dataset = dataset.batch(batch_size, drop_remainder=True).prefetch(1)\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)"
] ]
}, },
{ {