首页 > Web开发 > 详细

pytorch--报错 AttributeError: ‘Net’ object has no attribute ‘copy’

时间:2021-05-08 16:27:30      阅读:11      评论:0      收藏:0      [点我收藏+]

报错信息: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’

原文:https://www.cnblogs.com/zukang/p/14743990.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!