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