pytorch调用inception-v3报错 'max() received an invalid combination of arguments'

发表于 2020-04-07 分类于 pytorch 本文字数： 88 阅读时长 ≈ 1 分钟

pytorch调用inception-v3报错：

error：
  train_model(model, criterion, optimizer, scheduler, num_epochs)
       37    outputs = model(inputs)
  ---> 38    _, preds = torch.max(outputs, 1)
       39    loss = criterion(outputs, labels)
  TypeError: max() received an invalid combination of arguments - got (InceptionOutputs, int), but expected one of: ...

软件版本：

python: 3.7.7
pytorch: 1.4.0
torchvision: 0.5.0

解决办法：

参考 https://stackoverflow.com/questions/51045839/pytorch-inceptionv3-transfer-learning-gives-error-max-received-an-invalid-co
1
2
3
model_ft = models.inception_v3(pretrained=True)
# 加入下面这一行
model_ft.aux_logits=False