Thanks a lot Tom, I am not sure I understand the last part of your answer - the lines you suggested will initialize the weights in the classifier part according to Xavier, but not the bias part, which is not stored as part of model.classifier.parameters() ? The bias is stored in model.features[-1].weight?
When using Xavier, it is common to init the bias to be 0? Or also by Xavier?
Hope you will be able to help me understand this part. Thanks a lot!