【PyTorch】使用pytorch提供的模型训练inception V3(使用预训练模型)

前言:

最近在用vgg16跑数据,但是参数量实在太大,跑的非常慢,虽然最后结果还算可观。后面又研究了一下GoogLeNet,自己写了inceptionV1的代码,因为跑vgg16一直占着电脑,所以就没有跑,。

在这期间,在小老弟电脑上跑了MobileNet V1,找不到pytorch版的预训练权重,所以从头开始跑,过拟合非常严重,等之后研究一下是什么原因,猜测是数据集不平衡太严重以及没有预训练。

今天直接调用了pytorch的inception V3的代码跑数据,对于这个网络还没有深入研究,所以跑的时候遇到了一些比较低级的问题,但是在百度上也搜不到,最后在Google上找到了解决办法。

问题:

'InceptionOuputs' object has no attribute 'log_softmax'

在Torch 0.4.0上报的错误应该是:

'tuple' object has no attribute 'log_softmax'

 解决方法:

inception系列都有一个辅助分类器,在训练的时候可以选择是否要用辅助分类器,我在训练的时候选择了使用,所以网络的输出是:

       if self.training and self.aux_logits:
           return _InceptionOuputs(x, aux)

 而_InceptionOuputs在代码

  • 7
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值