Solved by rajasekhar in post #15 
              
              
                For GoogLeNet, there are 2 aux branches. So we have to do this way: 
aux1, aux2, output = model(inputs)     
loss1 = criterion(outputs, target)
loss2 = criterion(aux1, target)
loss3 = criterion(aux2, target)
loss = loss1 + 0.3*(loss2+loss3)

For Inception v3, it has only one aux branch. 
outputs, au…