inception_v3
was added in March 2017, so I’m not sure, if we are talking about the same model.
Anyway, the Finetuning Tutorial has specific code paths for the inception model and for its auxiliary outputs.
Logits are basically the input to a softmax layer.
Since nn.CrossentropyLoss
is using F.log_softmax
and nn.NLLLoss
internally, you don’t need to apply softmax manually.