Exception has occurred: AttributeError ‘InceptionOutputs‘ object has no attribute ‘log_softmax‘

在使用InceptionV3进行训练时遇到AttributeError:'InceptionOutputs'对象没有属性log_softmax。原因是辅助分支aux_logits在评估状态时不使用,导致问题。修正方法是确保仅使用logits输出。

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

在使用 inception_v3 训练时报的错,当时使用的是如下代码:

. . .
optimizer.zero_grad()
outs_1 = model(imgs_1)
loss = criterion(outs_1,labels)
loss.backward()
optimizer.step()
. . .

就这样报错了,因为平时都是这样写的,没太注意。通过调试观察网络输出outs_1,发现有2种值,outs_1结果如下:

InceptionOutputs(logits=tensor([[-22.2189, -25.1008, -19.7691,   1.4552, -25.5531],
        [-28.8628
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值