AttributeError: 'tuple' object has no attribute 'log_softmax' when run inception_v3

[binbinmeng]

python main.py -a inception_v3 --epoch 5 -b 64 /workspace/imagenet/ --fp16
=> creating model 'inception_v3'
Traceback (most recent call last):
File "main.py", line 466, in
main()
File "main.py", line 212, in main
train(train_loader, model, criterion, optimizer, epoch)
File "main.py", line 293, in train
loss = criterion(output, target_var)
File "/opt/conda/envs/pytorch-py3.6/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call
result = self.forward(*input, **kwargs)
File "/opt/conda/envs/pytorch-py3.6/lib/python3.6/site-packages/torch/nn/modules/loss.py", line 779, in forward
self.ignore_index, self.reduce)
File "/opt/conda/envs/pytorch-py3.6/lib/python3.6/site-packages/torch/nn/functional.py", line 1454, in cross_entropy
return nll_loss(log_softmax(input, 1), target, weight, size_average, ignore_index, reduce)
File "/opt/conda/envs/pytorch-py3.6/lib/python3.6/site-packages/torch/nn/functional.py", line 946, in log_softmax
return input.log_softmax(dim)
AttributeError: 'tuple' object has no attribute 'log_softmax'

node:
In [2]: print(torch.version)
0.5.0a0

[mcarilli]

Thanks for letting me know, I'll see if I can reproduce this tomorrow.

[csarofeen]

This is known for inception_v3, please see pytorch/vision#302

[wlj567]

Traceback (most recent call last):
File "/home/pxg/DAN/DANet/experiments/segmentation/train.py", line 292, in
trainer.training(epoch)
File "/home/pxg/DAN/DANet/experiments/segmentation/train.py", line 214, in training
aux1, aux2, outputs = self.model(image)
ValueError: not enough values to unpack (expected 3, got 1)

Hello, I modified it according to the above method and still reported an error. Can you help me see it? Thank you very much.