Ask questionsUse torch.device() with torch.load(..., map_location=torch.device())
It will be useful to allow map_location to be an instance of torch.device for transferability. By now attempt to do so gives an error: TypeError: 'torch.Device' object is not callable
Answer questions cooleel
Nice. met the same issue with torch 0.4, and solved with the solution provided here. Thanks all.