Correcting the batchsize

lalessia · Nov 7, 2019 · 6b5ba9a · 6b5ba9a
1 parent 1b2ecb5
commit 6b5ba9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ch17/ch17_optional_DCGAN.ipynb b/ch17/ch17_optional_DCGAN.ipynb
@@ -699,7 +699,7 @@
    "source": [
     "\n",
     "num_epochs = 100\n",
-    "batch_size = 8\n",
+    "batch_size = 64\n",
     "image_size = (28, 28)\n",
     "z_size = 20\n",
     "mode_z = 'uniform'\n",
@@ -894,7 +894,7 @@
     "start_time = time.time()\n",
     "for epoch in range(1, num_epochs+1):\n",
     "    epoch_losses, epoch_d_vals = [], []\n",
-    "    for i,(input_z,input_real) in enumerate(mnist_trainset.take(4)): #.take(4)\n",
+    "    for i,(input_z,input_real) in enumerate(mnist_trainset): #.take(4)\n",
     "        \n",
     "        ## Compute generator's loss\n",
     "        with tf.GradientTape() as g_tape:\n",