Below are the image learners from torchvision:
An overview of all learners can also be found on the mlr-org website.
| key | label | task_type | feature_types | packages |
|---|---|---|---|---|
| classif.alexnet | AlexNet | classif | lazy_tensor | torchvision |
| classif.convnext_base_1k | ConvNeXt-B (ImageNet-1k) | classif | lazy_tensor | torchvision |
| classif.convnext_base_22k | ConvNeXt-B (ImageNet-22k) | classif | lazy_tensor | torchvision |
| classif.convnext_large_1k | ConvNeXt-L (ImageNet-1k) | classif | lazy_tensor | torchvision |
| classif.convnext_large_22k | ConvNeXt-L (ImageNet-22k) | classif | lazy_tensor | torchvision |
| classif.convnext_small_22k | ConvNeXt-S (ImageNet-22k) | classif | lazy_tensor | torchvision |
| classif.convnext_tiny_1k | ConvNeXt-T (ImageNet-1k) | classif | lazy_tensor | torchvision |
| classif.convnext_tiny_22k | ConvNeXt-T (ImageNet-22k) | classif | lazy_tensor | torchvision |
| classif.efficientnet_b0 | EfficientNet-B0 | classif | lazy_tensor | torchvision |
| classif.efficientnet_b1 | EfficientNet-B1 | classif | lazy_tensor | torchvision |
| classif.efficientnet_b2 | EfficientNet-B2 | classif | lazy_tensor | torchvision |
| classif.efficientnet_b3 | EfficientNet-B3 | classif | lazy_tensor | torchvision |
| classif.efficientnet_b4 | EfficientNet-B4 | classif | lazy_tensor | torchvision |
| classif.efficientnet_b5 | EfficientNet-B5 | classif | lazy_tensor | torchvision |
| classif.efficientnet_b6 | EfficientNet-B6 | classif | lazy_tensor | torchvision |
| classif.efficientnet_b7 | EfficientNet-B7 | classif | lazy_tensor | torchvision |
| classif.efficientnet_v2_l | EfficientNetV2-L | classif | lazy_tensor | torchvision |
| classif.efficientnet_v2_m | EfficientNetV2-M | classif | lazy_tensor | torchvision |
| classif.efficientnet_v2_s | EfficientNetV2-S | classif | lazy_tensor | torchvision |
| classif.inception_v3 | Inception v3 | classif | lazy_tensor | torchvision |
| classif.maxvit | MaxViT-T | classif | lazy_tensor | torchvision |
| classif.mobilenet_v2 | MobileNetV2 | classif | lazy_tensor | torchvision |
| classif.mobilenet_v3_large | MobileNetV3-Large | classif | lazy_tensor | torchvision |
| classif.mobilenet_v3_small | MobileNetV3-Small | classif | lazy_tensor | torchvision |
| classif.resnet101 | ResNet-101 | classif | lazy_tensor | torchvision |
| classif.resnet152 | ResNet-152 | classif | lazy_tensor | torchvision |
| classif.resnet18 | ResNet-18 | classif | lazy_tensor | torchvision |
| classif.resnet34 | ResNet-34 | classif | lazy_tensor | torchvision |
| classif.resnet50 | ResNet-50 | classif | lazy_tensor | torchvision |
| classif.resnext101_32x8d | ResNeXt-101 32x8d | classif | lazy_tensor | torchvision |
| classif.resnext50_32x4d | ResNeXt-50 32x4d | classif | lazy_tensor | torchvision |
| classif.vgg11 | VGG-11 | classif | lazy_tensor | torchvision |
| classif.vgg11_bn | VGG-11 with batch normalization | classif | lazy_tensor | torchvision |
| classif.vgg13 | VGG-13 | classif | lazy_tensor | torchvision |
| classif.vgg13_bn | VGG-13 with batch normalization | classif | lazy_tensor | torchvision |
| classif.vgg16 | VGG-16 | classif | lazy_tensor | torchvision |
| classif.vgg16_bn | VGG-16 with batch normalization | classif | lazy_tensor | torchvision |
| classif.vgg19 | VGG-19 | classif | lazy_tensor | torchvision |
| classif.vgg19_bn | VGG-19 with batch normalization | classif | lazy_tensor | torchvision |
| classif.vit_b_16 | ViT-B/16 | classif | lazy_tensor | torchvision |
| classif.vit_b_32 | ViT-B/32 | classif | lazy_tensor | torchvision |
| classif.vit_h_14 | ViT-H/14 | classif | lazy_tensor | torchvision |
| classif.vit_l_16 | ViT-L/16 | classif | lazy_tensor | torchvision |
| classif.vit_l_32 | ViT-L/32 | classif | lazy_tensor | torchvision |
| classif.wide_resnet101_2 | Wide ResNet-101-2 | classif | lazy_tensor | torchvision |
| classif.wide_resnet50_2 | Wide ResNet-50-2 | classif | lazy_tensor | torchvision |