报错信息:AttributeError: ‘Net’ object has no attribute ‘copy’
分析:报错是发生在加载预训练模型时,很可能时两种pytorch加载预训练模型方式弄混了。
解决:按照下面思路加载预训练模型就好。
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)
pytorch--报错 AttributeError: ‘Net’ object has no attribute ‘copy’