@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