From 90a8e316e2c962b655431682d1ea1687fe4da96f Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <sylvain.gugger@gmail.com>
Date: Sat, 21 Mar 2020 08:38:44 -0700
Subject: [PATCH] Update

---
 01_intro.ipynb       | 18 ++++++++++++++++++
 clean/01_intro.ipynb | 17 +++++++++++++++++
 2 files changed, 35 insertions(+)

diff --git a/01_intro.ipynb b/01_intro.ipynb
index e9b2744..73cc2f4 100644
--- a/01_intro.ipynb
+++ b/01_intro.ipynb
@@ -2168,6 +2168,24 @@
     "learn.fine_tune(4, 1e-2)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "#clean\n",
+    "If you hit a \"CUDA out of memory error\" after running this cell, click on the menu Kernel, then restart. Instead of executing the cell above, copy and paste the following code in it:\n",
+    "\n",
+    "```\n",
+    "from fastai2.text.all import *\n",
+    "\n",
+    "dls = TextDataLoaders.from_folder(untar_data(URLs.IMDB), valid='test', bs=32)\n",
+    "learn = text_classifier_learner(dls, AWD_LSTM, drop_mult=0.5, metrics=accuracy)\n",
+    "learn.fine_tune(4, 1e-2)\n",
+    "```\n",
+    "\n",
+    "This reduces the batch size to 32 (we will explain this later). If you keep hitting the same error, change 32 by 16."
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
diff --git a/clean/01_intro.ipynb b/clean/01_intro.ipynb
index 9bf7c8d..c9ad32e 100644
--- a/clean/01_intro.ipynb
+++ b/clean/01_intro.ipynb
@@ -1072,6 +1072,23 @@
     "learn.fine_tune(4, 1e-2)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "If you hit a \"CUDA out of memory error\" after running this cell, click on the menu Kernel, then restart. Instead of executing the cell above, copy and paste the following code in it:\n",
+    "\n",
+    "```\n",
+    "from fastai2.text.all import *\n",
+    "\n",
+    "dls = TextDataLoaders.from_folder(untar_data(URLs.IMDB), valid='test', bs=32)\n",
+    "learn = text_classifier_learner(dls, AWD_LSTM, drop_mult=0.5, metrics=accuracy)\n",
+    "learn.fine_tune(4, 1e-2)\n",
+    "```\n",
+    "\n",
+    "This reduces the batch size to 32 (we will explain this later). If you keep hitting the same error, change 32 by 16."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,