Skip at the end

fastai · Feb 7, 2020 · 0ea6d11 · 0ea6d11
1 parent 222d2be
commit 0ea6d11
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/fastai2/callback/schedule.py b/fastai2/callback/schedule.py
@@ -198,7 +198,7 @@ def lr_find(self:Learner, start_lr=1e-7, end_lr=10, num_it=100, stop_div=True, s
     with self.no_logging(): self.fit(n_epoch, cbs=cb)
     if show_plot: self.recorder.plot_lr_find()
     if suggestions:
-        lrs,losses = tensor(self.recorder.lrs[5:]),tensor(self.recorder.losses[5:])
+        lrs,losses = tensor(self.recorder.lrs[5:-5]),tensor(self.recorder.losses[5:-5])
         lr_min = lrs[losses.argmin()].item()
         grads = (losses[1:]-losses[:-1]) / (lrs[1:].log()-lrs[:-1].log())
         lr_steep = lrs[grads.argmin()].item()

diff --git a/nbs/14_callback.schedule.ipynb b/nbs/14_callback.schedule.ipynb
@@ -1162,7 +1162,7 @@
     "    with self.no_logging(): self.fit(n_epoch, cbs=cb)\n",
     "    if show_plot: self.recorder.plot_lr_find()\n",
     "    if suggestions:\n",
-    "        lrs,losses = tensor(self.recorder.lrs[5:]),tensor(self.recorder.losses[5:])\n",
+    "        lrs,losses = tensor(self.recorder.lrs[5:-5]),tensor(self.recorder.losses[5:-5])\n",
     "        lr_min = lrs[losses.argmin()].item()\n",
     "        grads = (losses[1:]-losses[:-1]) / (lrs[1:].log()-lrs[:-1].log())\n",
     "        lr_steep = lrs[grads.argmin()].item()\n",

diff --git a/nbs/course/lesson3-imdb.ipynb b/nbs/course/lesson3-imdb.ipynb
@@ -582,7 +582,6 @@
    "outputs": [],
    "source": [
     "imdb_lm = DataBlock(blocks=(TextBlock.from_folder(path, is_lm=True),),\n",
-    "                    get_x=read_tokenized_file,\n",
     "                    get_items=partial(get_text_files, folders=['train', 'test', 'unsup']),\n",
     "                    splitter=RandomSplitter(0.1))\n",
     "\n",