@ishanic and @alphadl: Have either of you resolved this issue? I have having the same problem when I train my model on a GPU but then try to load it on my CPU for evaluation purposes.
The code I’m running is:
torch.load('model.pkl', map_location=lambda storage, loc: storage)

The error I got is:
*** RuntimeError: storage has wrong size: expected 857934349 got 128