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.mobilenet_v2 | Mobilenet V2 | 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 | classif | lazy_tensor | torchvision |
| classif.vgg13 | VGG 13 | classif | lazy_tensor | torchvision |
| classif.vgg13_bn | VGG 13 | classif | lazy_tensor | torchvision |
| classif.vgg16 | VGG 16 | classif | lazy_tensor | torchvision |
| classif.vgg16_bn | VGG 16 | classif | lazy_tensor | torchvision |
| classif.vgg19 | VGG 19 | classif | lazy_tensor | torchvision |
| classif.vgg19_bn | VGG 19 | classif | lazy_tensor | torchvision |