this section will help you (yes we have a big modelzoo)
http://pytorch.org/docs/torchvision/models.html 928
Also, this thread

  
    
    
    How to perform finetuning in Pytorch? 
  
  
    Can anyone tell me how to do finetuning in pytorch? Suppose, I have loaded the Resnet 18 pretrained model. Now I want to finetune it on my own dataset which contain say 10 classes. How to remove the last output layer and change to as per my requirement?