[tune] missing torch.load in mnist_pytorch_trainable.py (#5103)

haje01 · richardliaw · commit 4b56a5eb2704 · 2019-07-06T00:14:41.000-07:00
diff --git a/python/ray/tune/examples/mnist_pytorch_trainable.py b/python/ray/tune/examples/mnist_pytorch_trainable.py
@@ -167,7 +167,7 @@ def _save(self, checkpoint_dir):
         return checkpoint_path
 
     def _restore(self, checkpoint_path):
-        self.model.load_state_dict(checkpoint_path)
+        self.model.load_state_dict(torch.load(checkpoint_path))
 
 
 if __name__ == "__main__":
