From 4b56a5eb2704f5c42c39cbe4da6b76a03d71b9a6 Mon Sep 17 00:00:00 2001 From: Kim Jeong Ju Date: Sat, 6 Jul 2019 16:14:41 +0900 Subject: [PATCH] [tune] missing torch.load in mnist_pytorch_trainable.py (#5103) --- python/ray/tune/examples/mnist_pytorch_trainable.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/tune/examples/mnist_pytorch_trainable.py b/python/ray/tune/examples/mnist_pytorch_trainable.py index 741c5ef8c..7d92ba1c5 100644 --- a/python/ray/tune/examples/mnist_pytorch_trainable.py +++ b/python/ray/tune/examples/mnist_pytorch_trainable.py @@ -167,7 +167,7 @@ class TrainMNIST(Trainable): 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__":