For different models we reshape the network and reinitialize the classifier’s  layer, the link here “Finetuning Torchvision Models — PyTorch Tutorials 1.2.0 documentation 1” does not include Shufflenet V2. Does anyone have any idea??