Not able to train on imagenet using the inception architecture

Basically it is because of the two outputs returned in the output here. Since inception-v3 returns auxiliary classifier’s loss as well. Saw similar discussion on another thread still not clear how to incorporate the changes.