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

本文介绍了在PyTorch中使用Inception V3模型进行训练时遇到的'InceptionOuputs' object has no attribute 'log_softmax'错误,对比了0.4.0版本和新版本的差异,并提供了问题的解决方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

前言:

最近在用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在代码

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值