Imagenet example with inception v3

vision
3

@fmassa There seems to be another problem with customizing Inception. Even if the image size is adjusted looks like the Inception output is a tuple so the code to compute loss fails.

Code for loss computation:

output = model(input_var)
loss = criterion(output, target_var)

RuntimeError: expected a Variable argument, but got tuple

Looking at the inception implementation it seems like the following lines are being executed:

if self.training and self.aux_logits:
            return x, aux

Do you know what the aux_logits parameter is refering to? Should it simply be ignored for loss computation?

thanks

1 Like