On a cpu device, how to load checkpoint saved on gpu device

[apaszke (Adam Paszke)]

I’m sorry, my bad. This should work:

torch.load('my_file.pt', map_location=lambda storage, loc: storage)