https://blog.csdn.net/bc521bc/article/details/85623515
由于pytorch的模型保存的时候记录了device信息,所以加载的时候需要手动指定设备。
通过:, map_location={'cuda:1':'cuda:0'}
的写法
https://blog.csdn.net/bc521bc/article/details/85623515
由于pytorch的模型保存的时候记录了device信息,所以加载的时候需要手动指定设备。
通过:, map_location={'cuda:1':'cuda:0'}
的写法