读参数时,用map_location,gpu 1 -> gpu 0
torch.load('modelparameters.pth', map_location={'cuda:1':'cuda:0'})
参考:pytorch cpu与gpu load时相互转化 torch.load(map_location=)_bc521bc的博客-CSDN博客
读参数时,用map_location,gpu 1 -> gpu 0
torch.load('modelparameters.pth', map_location={'cuda:1':'cuda:0'})
参考:pytorch cpu与gpu load时相互转化 torch.load(map_location=)_bc521bc的博客-CSDN博客