Deep Learning, Pytorch, Fine Tuning, Computer Vision, Transfer Learning.
requiring modification. Source: kakaocdn.net By printing the model, you can see its architecture and identify the appropriate layer to modify. Look for a linear or FC layer that serves as the final classification layer, and replace it with a new layer that matches the number of classes or your task