arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-14 12:14:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add missing argument n_epochs=50

Browse Source
This commit is contained in:
Aurélien Geron
2022-04-01 12:11:37 +13:00
parent 05e891c997
commit 541ea1d7b1
1 changed files with 1 additions and 1 deletions
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 = 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)"
]
},
{