torchvision.models — PyTorch 1.7.0 documentation

torchvision.models¶ The models subpackage contains definitions of models for addressing different tasks, including: image classification, pixelwise semantic segmentation, object detection, instance segmentation, person keypoint detection and video classification. Classification¶ The models subpackage contains definitions for the following model architectures for image classification: AlexNet VGG R