Tiny fix

Hvass-Labs · Hvass-Labs · commit ae4aaa1aa7b9 · 2019-02-03T15:54:32.000+01:00
diff --git a/22_Image_Captioning.ipynb b/22_Image_Captioning.ipynb
@@ -70,6 +70,7 @@
     "import matplotlib.pyplot as plt\n",
     "import tensorflow as tf\n",
     "import numpy as np\n",
+    "import sys\n",
     "import os\n",
     "from PIL import Image\n",
     "from cache import cache"
@@ -1390,7 +1391,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "batch_size = 1024"
+    "batch_size = 512"
    ]
   },
   {
@@ -2452,7 +2453,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.1"
+   "version": "3.6.6"
   }
  },
  "nbformat": 4,
