ipynb typo

2018-05-05 17:30:08 -07:00 · 2018-05-05 17:30:08 -07:00 · 4ac6ce9ab5
parent c67ca6531e
commit 4ac6ce9ab5
1 changed files with 1 additions and 1 deletions
--- a/inference.ipynb
+++ b/inference.ipynb
@ -101,7 +101,7 @@
    "try:\n",
    "    model = model.module\n",
    "except:\n",
-    "    pass\n"
+    "    pass\n",
    "model.load_state_dict({k.replace('module.',''):v for k,v in torch.load(checkpoint_path)['state_dict'].items()})\n",
    "_ = model.eval()"
   ]