出错原因
显卡设备不匹配,原为'cuda:1',现为'cuda:0'。
解决方案
原代码:
model.load_state_dict(model_path)
现代码:
loaded_state = torch.load(model_path, map_location='cuda:0')
model.load_state_dict(loaded_state)
显卡设备不匹配,原为'cuda:1',现为'cuda:0'。
原代码:
model.load_state_dict(model_path)
现代码:
loaded_state = torch.load(model_path, map_location='cuda:0')
model.load_state_dict(loaded_state)