python - 属性错误 : 'tuple' object has no attribute 'log_softmax'

Question

同时尝试通过更改最后一个 fc 层来为我自己的数据集微调 inception_V3

last_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'

Answer 1

这是一个众所周知的问题。

尝试以下解决方案之一:

1. 通过同时将 aux_logits=False 传递给 inception_v3 函数，在此处创建模型时禁用 aux_logits。

2. 编辑您的训练函数以接受并解压缩返回的元组，使其类似于:output, aux = model(input_var)

检查以下 link了解更多信息。