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