错误-torch.nn.modules.module.ModuleAttributeError:: ‘xxx’ object has no attribute ‘copy’

使用torch.load_state_dict()出现的问题

model.load_state_dict(torch.load(model_path))

如下报错：

torch.nn.modules.module.ModuleAttributeError: 'XXXNet' object has no attribute 'copy'

发现问题出现在训练模型的代码中，在保存模型时，使用的是

	torch.save(model, path)

应该使用

	torch.save(model.state_dict(), path)