AttributeError: ‘Net’ object has no attribute ‘copy’ 模型如何保存就如何加载
解决:按照下面思路加载预训练模型就好。
1.保存加载state_dict方式(推荐)
保存:torch.save(model.state_dict(), PATH) # 推荐的文件后缀名是pt或pth
加载:model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
2.保存加载整个模型
保存:torch.save(model, PATH)
加载:model = torch.load(PATH)