Inception_v3 needs more than a single sample during training as at some point inside the model the activation will have the shape [batch_size, 768, 1, 1] and thus the batchnorm layer won’t be able to calculate the batch statistics.
You could set the model to eval(), which will use the running statistics instead or increase the batch size.