Finetuning : need to modify model to go from 1000 to 2 output classes?

[nielsr]

Hi,

Yes this warning is fine. It tells you that you are loading all the pre-trained weights, however the classifier on top has its weights and bias randomly initialized. Hence it needs to be fine-tuned on a custom dataset.