pytorch加载模型报错TypeError: ‘collections.OrderedDict‘ object is not callable

参考博客:添加链接描述

  • 错误代码
    ----之前训练了一个网络模型,保存了网络训练参数,代码如下
torch.save(module1.state_dict(),"module1_{}.pth".format(i))
  • 错误原因:
    ----在进行模型测试的时候,使用如下代码对训练模型直接进行加载:
model=torch.load("Pytorch/module1_9.pth")
  • 改正方法
    ----先载入网络结构,再导入网络的参数( Module1是我的网络模型)
model = Module1()	# 导入网络结构
model.load_state_dict(torch.load("Pytorch/module1_9.pth")) # 导入网络的参数
  • 另外
    ----保存模型的两种方式:保存整个训练模型、只保存模型参数
 # 保存整个模型
 torch.save(module,"module_{}.pth".format(i))
 # 只保存模型参数(官方推荐,并且占内存小)
 torch.save(module.state_dict(),"module_{}.pth".format(i))
  • 6
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值