torch.load加载模型后出现问题

写了两篇论文,做实验时候都遇到了使用torch.load加载模型时的问题,特此记录,欢迎交流。

1.保存整个模型

torch.save(model,'net.pth')

这种保存方式很局限,我在自己电脑上训练时候会使用这种方式(因为懒惰,哈哈哈哈~)

出现的问题:

1.1 显示没有xxmodule

这种很常见,只要你保存模型后发生过以下情况:

        1.1.1更改过模型结构

        1.1.2导入别的类时这个类更改过或者相对路径放生改变(一定检查路径!!!)

1.2 能运行就是效果非常差

这个问题我搜了两天都没找到方法,后来自己灵光一现解决了!!!

一句话:去检查TestDataset!!!拿我来讲,我训练时候的Test是使用了normalize的,后来觉得没有用就注释了,结果出现这种问题。

2.保存参数

torch.save(model.state_dict(),'net_params.pth')

这种挺好用的,我在服务器上训练使用这种。方便下载下来保存验证。(毕竟服务器太贵了,呜呜)

这种我目前没发生什么问题,但是看了一些帖子提到了。

2.1运行不通

主要是多GPU和单GPU问题,具体大家百度一下~

2.2效果很差

验证前,没写这句话

model.eval()

其他问题,欢迎交流和指正,第一次写CSDN,献给这个奇葩的Bug!!!

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值