Pass device to torch.load instead of map_location function

Open daniel-j-h opened this issue 7 years ago • 0 comments

See for context: https://github.com/pytorch/pytorch/issues/7178. We currently are passing devices around except for torch.load where we have to pass a map_location function argument.

In https://github.com/pytorch/pytorch/pull/7339 this got changed and we can now pass the device to the torch.load function.

Note: 0.4.0 does not yet include these changes; we need to wait for another feature release.

Tasks:

[ ] Pass device to load function
[ ] Test and merge in sync with next PyTorch release

Jun 10 '18 23:06 daniel-j-h