torch.load()报错torch.nn.modules.module.ModuleAttributeError: ‘mlp’ object has no attribute 'copy’的原因:
模型保存与加载的格式不一致
解决方法:
方法一:
保存整个模型:
torch.save(model,'save.pt')
#加载模型
torch.load('save.pt')
方法二:
保存训练好的权重:
torch.save(model.state_dict(), 'save.pt')
#加载模型
model.load_state_dict(torch.load("save.pt"))