在使用 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

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

被折叠的 条评论
为什么被折叠?



