The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.
By using the torchvision package, we could easily load and use the pretrained vision models such as Resnet and Alexnet.
Example notebook for using pretrained models with torchvision