or
用cuda版本号为5训练,用cuda版本号为0测试:
在加载模型时,torch.load("*.pth",map_location={‘cuda:5’: ‘cuda:0’})即可。
pytorch加载预训练模型,去module;测试时更改cuda版本号
最新推荐文章于 2022-11-10 16:51:23 发布
or
用cuda版本号为5训练,用cuda版本号为0测试:
在加载模型时,torch.load("*.pth",map_location={‘cuda:5’: ‘cuda:0’})即可。