pytorch inception v3 KeyError: <class 'tuple'>解决方法

刚入门深度学习，试着跑了一下resnet18和resnet50都没有问题，但是在运行inception v3的时候遇到一个问题怎么都解决不了

错误输出如下：

Traceback (most recent call last):
  File "transfer_learning_tutorial.py", line 277, in <module>
    num_epochs=25)
  File "transfer_learning_tutorial.py", line 179, in train_model
    _, preds = torch.max(outputs, 1)
TypeError: max() received an invalid combination of arguments - got (tuple, int), but expected one of:
 * (Tensor input)
 * (Tensor input, Tensor other, Tensor out)
 * (Tensor input, int dim, bool keepdim, tuple of Tensors out)

在github上找到答案

inception源码连接第125行，在train模式下并且aux_logits打开的情况下，返回x, aux。

所以解决方法有以下两个：

方法1.创建inception模型的时候，关闭aux_logits。设置关键字参数aux_logits=False

方法2.接收返回的aux。

output, aux = model(input_var)#注意只有在训练模式下接收两个参数
out=model(input_var)#在求值模式下，仍然只返回一个参数

                    if phase=='train':
                        outputs,aux= model(inputs)
                    else:
                        outputs=model(inputs)