I see.  Do you have an idea of what specifically needs to be adapted to make it work?  I’m not familiar with the exact computation of loss in the inception network so I’m not sure how to change the pytorch code.
Thanks.