Thanks @ptrblck I tried fine tuning the classification layer but it doesn’t provide good results so far…I’ll experiment with fine tuning the entire head    I notice there is not a pre trained model for the resnet50 case in torchvision. Am I right in saying I can just take the first half of resnet101?