同时尝试通过更改最后一个fc层(例如
)来调整自己的数据集的inception_V3last_layer =nn.Linear(n_inputs, len(classes))
inception_v3.fc = last_layer
在那之后,当我训练它时,这个位置出现了这个错误
# on training loop
output = inception_v3(data)
# calculate the batch loss
loss = criterion(output, target)
错误是
AttributeError: 'tuple' object has no attribute 'log_softmax'
答案 0 :(得分:3)
这是一个众所周知的问题。
尝试以下解决方案之一:
在此处创建模型时也通过将aux_logits=False
传递给inception_v3函数来禁用aux_logits。
编辑您的火车函数,以接受并解压缩返回的元组,例如:output, aux = model(input_var)
检查以下link以获得更多信息。
答案 1 :(得分:0)