From 4ac6ce9ab51f43bbb21993e995b87e4db13f02e4 Mon Sep 17 00:00:00 2001 From: Raul Puri Date: Sat, 5 May 2018 17:30:08 -0700 Subject: [PATCH] ipynb typo --- inference.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference.ipynb b/inference.ipynb index 2e94ff3..2aa9b50 100644 --- 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()" ]