wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-08 03:30:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Kim Jeong Ju
2019-07-06 16:14:41 +09:00
committed by Richard Liaw
parent c5253cc300
commit 4b56a5eb27
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/examples/mnist_pytorch_trainable.py
+1 -1
View File
@@ -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__":