From 6b5ba9a19de72cf0811f0481b1d80fe9f98a3a02 Mon Sep 17 00:00:00 2001
From: vmirly <mirjalil@msu.edu>
Date: Wed, 6 Nov 2019 20:47:57 -0500
Subject: [PATCH] Correcting the batchsize

---
 ch17/ch17_optional_DCGAN.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ch17/ch17_optional_DCGAN.ipynb b/ch17/ch17_optional_DCGAN.ipynb
index 2f31fab1..fe696325 100644
--- 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",