wassname/ray
1
0 Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-10-28 13:21:06 +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 authored and Richard Liaw committed 2019-07-06 00:14:41 -07:00
1 parent c5253cc300
commit 4b56a5eb27
1 file changed
+1 -1
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__":