fix fastai/fastbook/issues/263

potential fix for fastai/fastbook#263
Tagar · Sep 14, 2020 · 92d8a92 · 92d8a92
1 parent 760b54e
commit 92d8a92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbs/14_callback.schedule.ipynb b/nbs/14_callback.schedule.ipynb
@@ -415,7 +415,7 @@
     "    assert torch.all(pcts >= 0)\n",
     "    pcts = torch.cumsum(pcts, 0)\n",
     "    def _inner(pos):\n",
-    "        if pos == 1.: return scheds[-1](1.)\n",
+    "        if int(pos) == 1: return scheds[-1](1.)\n",
     "        idx = (pos >= pcts).nonzero().max()\n",
     "        actual_pos = (pos-pcts[idx]) / (pcts[idx+1]-pcts[idx])\n",
     "        return scheds[idx](actual_pos.item())\n",