I’m trying to train a classifier on 15k images over five categories using googlenet architecture.
I followed the fine-tune tutorial (but used as pretrained=false just to train from scratch).
But the training is only possible if i set the ‘aux logits as false’
‘’’
model.aux.logits=False
‘’’
Can someone explain why I have to do this for training?