Intel® Transfer Learning Tool Supported Models
April 8, 2024 · View on GitHub
Image Classification
| Model name | Framework | Model Hub |
|---|---|---|
| alexnet | PyTorch* | Torchvision* |
| convnext_base | PyTorch | Torchvision |
| convnext_large | PyTorch | Torchvision |
| convnext_small | PyTorch | Torchvision |
| convnext_tiny | PyTorch | Torchvision |
| densenet121 | PyTorch | Torchvision |
| densenet161 | PyTorch | Torchvision |
| densenet169 | PyTorch | Torchvision |
| densenet201 | PyTorch | Torchvision |
| efficientnetv2-b0 | TensorFlow* | TensorFlow Hub* |
| efficientnetv2-b1 | TensorFlow | TensorFlow Hub |
| efficientnetv2-b2 | TensorFlow | TensorFlow Hub |
| efficientnetv2-b3 | TensorFlow | TensorFlow Hub |
| efficientnetv2-s | TensorFlow | TensorFlow Hub |
| efficientnet_b0 | TensorFlow | TensorFlow Hub |
| efficientnet_b0 | PyTorch | Torchvision |
| efficientnet_b1 | TensorFlow | TensorFlow Hub |
| efficientnet_b1 | PyTorch | Torchvision |
| efficientnet_b2 | TensorFlow | TensorFlow Hub |
| efficientnet_b2 | PyTorch | Torchvision |
| efficientnet_b3 | TensorFlow | TensorFlow Hub |
| efficientnet_b3 | PyTorch | Torchvision |
| efficientnet_b4 | TensorFlow | TensorFlow Hub |
| efficientnet_b4 | PyTorch | Torchvision |
| efficientnet_b5 | TensorFlow | TensorFlow Hub |
| efficientnet_b5 | PyTorch | Torchvision |
| efficientnet_b6 | TensorFlow | TensorFlow Hub |
| efficientnet_b6 | PyTorch | Torchvision |
| efficientnet_b7 | TensorFlow | TensorFlow Hub |
| efficientnet_b7 | PyTorch | Torchvision |
| googlenet | PyTorch | Torchvision |
| inception_v3 | TensorFlow | TensorFlow Hub |
| mnasnet0_5 | PyTorch | Torchvision |
| mnasnet1_0 | PyTorch | Torchvision |
| mobilenet_v2 | PyTorch | Torchvision |
| mobilenet_v2_100_224 | TensorFlow | TensorFlow Hub |
| mobilenet_v3_large | PyTorch | Torchvision |
| mobilenet_v3_small | PyTorch | Torchvision |
| nasnet_large | TensorFlow | TensorFlow Hub |
| proxyless_cpu | PyTorch | PyTorch Hub* |
| regnet_x_16gf | PyTorch | Torchvision |
| regnet_x_1_6gf | PyTorch | Torchvision |
| regnet_x_32gf | PyTorch | Torchvision |
| regnet_x_3_2gf | PyTorch | Torchvision |
| regnet_x_400mf | PyTorch | Torchvision |
| regnet_x_800mf | PyTorch | Torchvision |
| regnet_x_8gf | PyTorch | Torchvision |
| regnet_y_16gf | PyTorch | Torchvision |
| regnet_y_1_6gf | PyTorch | Torchvision |
| regnet_y_32gf | PyTorch | Torchvision |
| regnet_y_3_2gf | PyTorch | Torchvision |
| regnet_y_400mf | PyTorch | Torchvision |
| regnet_y_800mf | PyTorch | Torchvision |
| regnet_y_8gf | PyTorch | Torchvision |
| resnet101 | PyTorch | Torchvision |
| resnet152 | PyTorch | Torchvision |
| resnet18 | PyTorch | Torchvision |
| resnet18_ssl | PyTorch | PyTorch Hub |
| resnet18_swsl | PyTorch | PyTorch Hub |
| resnet34 | PyTorch | Torchvision |
| resnet50 | PyTorch | Torchvision |
| resnet50_ssl | PyTorch | PyTorch Hub |
| resnet50_swsl | PyTorch | PyTorch Hub |
| resnet_v1_50 | TensorFlow | TensorFlow Hub |
| resnet_v2_101 | TensorFlow | TensorFlow Hub |
| resnet_v2_50 | TensorFlow | TensorFlow Hub |
| resnext101_32x16d_ssl | PyTorch | PyTorch Hub |
| resnext101_32x16d_swsl | PyTorch | PyTorch Hub |
| resnext101_32x16d_wsl | PyTorch | PyTorch Hub |
| resnext101_32x32d_wsl | PyTorch | PyTorch Hub |
| resnext101_32x48d_wsl | PyTorch | PyTorch Hub |
| resnext101_32x4d_ssl | PyTorch | PyTorch Hub |
| resnext101_32x4d_swsl | PyTorch | PyTorch Hub |
| resnext101_32x8d | PyTorch | Torchvision |
| resnext101_32x8d_ssl | PyTorch | PyTorch Hub |
| resnext101_32x8d_swsl | PyTorch | PyTorch Hub |
| resnext101_32x8d_wsl | PyTorch | PyTorch Hub |
| resnext50_32x4d | PyTorch | Torchvision |
| resnext50_32x4d_ssl | PyTorch | PyTorch Hub |
| resnext50_32x4d_swsl | PyTorch | PyTorch Hub |
| shufflenet_v2_x0_5 | PyTorch | Torchvision |
| shufflenet_v2_x1_0 | PyTorch | Torchvision |
| vgg11 | PyTorch | Torchvision |
| vgg11_bn | PyTorch | Torchvision |
| vgg13 | PyTorch | Torchvision |
| vgg13_bn | PyTorch | Torchvision |
| vgg16 | PyTorch | Torchvision |
| vgg16_bn | PyTorch | Torchvision |
| vgg19 | PyTorch | Torchvision |
| vgg19_bn | PyTorch | Torchvision |
| vit_b_16 | PyTorch | Torchvision |
| vit_b_32 | PyTorch | Torchvision |
| vit_l_16 | PyTorch | Torchvision |
| vit_l_32 | PyTorch | Torchvision |
| wide_resnet101_2 | PyTorch | Torchvision |
| wide_resnet50_2 | PyTorch | Torchvision |
| ConvNeXtBase | TensorFlow | Keras* |
| ConvNeXtLarge | TensorFlow | Keras |
| ConvNeXtSmall | TensorFlow | Keras |
| ConvNeXtTiny | TensorFlow | Keras |
| ConvNeXtXLarge | TensorFlow | Keras |
| DenseNet121 | TensorFlow | Keras |
| DenseNet169 | TensorFlow | Keras |
| DenseNet201 | TensorFlow | Keras |
| EfficientNetV2B0 | TensorFlow | Keras |
| EfficientNetV2B1 | TensorFlow | Keras |
| EfficientNetV2B2 | TensorFlow | Keras |
| EfficientNetV2B3 | TensorFlow | Keras |
| EfficientNetV2L | TensorFlow | Keras |
| EfficientNetV2M | TensorFlow | Keras |
| EfficientNetV2S | TensorFlow | Keras |
| InceptionResNetV2 | TensorFlow | Keras |
| InceptionV3 | TensorFlow | Keras |
| MobileNet | TensorFlow | Keras |
| MobileNetV2 | TensorFlow | Keras |
| NASNetLarge | TensorFlow | Keras |
| NASNetMobile | TensorFlow | Keras |
| ResNet101 | TensorFlow | Keras |
| ResNet101V2 | TensorFlow | Keras |
| ResNet152 | TensorFlow | Keras |
| ResNet152V2 | TensorFlow | Keras |
| ResNet50 | TensorFlow | Keras |
| ResNet50V2 | TensorFlow | Keras |
| VGG16 | TensorFlow | Keras |
| VGG19 | TensorFlow | Keras |
| Xception | TensorFlow | Keras |
Text Classification
| Model name | Framework | Model Hub |
|---|---|---|
| bert-base-cased | PyTorch | Hugging Face* |
| bert-base-uncased | TensorFlow | Hugging Face |
| bert-large-uncased | TensorFlow | Hugging Face |
| bert-large-uncased | PyTorch | Hugging Face |
| clinical-bert | PyTorch | Hugging Face |
| distilbert-base-uncased | PyTorch | Hugging Face |
| google_bert_uncased_L-10_H-128_A-2 | TensorFlow | Hugging Face |
| google_bert_uncased_L-10_H-256_A-4 | TensorFlow | Hugging Face |
| google_bert_uncased_L-10_H-512_A-8 | TensorFlow | Hugging Face |
| google_bert_uncased_L-10_H-768_A-12 | TensorFlow | Hugging Face |
| google_bert_uncased_L-12_H-128_A-2 | TensorFlow | Hugging Face |
| google_bert_uncased_L-12_H-256_A-4 | TensorFlow | Hugging Face |
| google_bert_uncased_L-12_H-512_A-8 | TensorFlow | Hugging Face |
| google_bert_uncased_L-12_H-768_A-12 | TensorFlow | Hugging Face |
| google_bert_uncased_L-2_H-128_A-2 | TensorFlow | Hugging Face |
| google_bert_uncased_L-2_H-256_A-4 | TensorFlow | Hugging Face |
| google_bert_uncased_L-2_H-512_A-8 | TensorFlow | Hugging Face |
| google_bert_uncased_L-2_H-768_A-12 | TensorFlow | Hugging Face |
| google_bert_uncased_L-4_H-128_A-2 | TensorFlow | Hugging Face |
| google_bert_uncased_L-4_H-256_A-4 | TensorFlow | Hugging Face |
| google_bert_uncased_L-4_H-512_A-8 | TensorFlow | Hugging Face |
| google_bert_uncased_L-4_H-768_A-12 | TensorFlow | Hugging Face |
| google_bert_uncased_L-6_H-128_A-2 | TensorFlow | Hugging Face |
| google_bert_uncased_L-6_H-256_A-4 | TensorFlow | Hugging Face |
| google_bert_uncased_L-6_H-512_A-8 | TensorFlow | Hugging Face |
| google_bert_uncased_L-6_H-768_A-12 | TensorFlow | Hugging Face |
| google_bert_uncased_L-8_H-128_A-2 | TensorFlow | Hugging Face |
| google_bert_uncased_L-8_H-256_A-4 | TensorFlow | Hugging Face |
| google_bert_uncased_L-8_H-512_A-8 | TensorFlow | Hugging Face |
| google_bert_uncased_L-8_H-768_A-12 | TensorFlow | Hugging Face |
Image Anomaly Detection
| Model name | Framework | Model Hub |
|---|---|---|
| alexnet | PyTorch | Torchvision |
| convnext_base | PyTorch | Torchvision |
| convnext_large | PyTorch | Torchvision |
| convnext_small | PyTorch | Torchvision |
| convnext_tiny | PyTorch | Torchvision |
| densenet121 | PyTorch | Torchvision |
| densenet161 | PyTorch | Torchvision |
| densenet169 | PyTorch | Torchvision |
| densenet201 | PyTorch | Torchvision |
| dinov2_vitb14 | PyTorch | PyTorch Hub |
| dinov2_vitb14_reg | PyTorch | PyTorch Hub |
| dinov2_vitg14 | PyTorch | PyTorch Hub |
| dinov2_vitg14_reg | PyTorch | PyTorch Hub |
| dinov2_vitl14 | PyTorch | PyTorch Hub |
| dinov2_vitl14_reg | PyTorch | PyTorch Hub |
| dinov2_vits14 | PyTorch | PyTorch Hub |
| dinov2_vits14_reg | PyTorch | PyTorch Hub |
| efficientnet_b0 | PyTorch | Torchvision |
| efficientnet_b1 | PyTorch | Torchvision |
| efficientnet_b2 | PyTorch | Torchvision |
| efficientnet_b3 | PyTorch | Torchvision |
| efficientnet_b4 | PyTorch | Torchvision |
| efficientnet_b5 | PyTorch | Torchvision |
| efficientnet_b6 | PyTorch | Torchvision |
| efficientnet_b7 | PyTorch | Torchvision |
| googlenet | PyTorch | Torchvision |
| mnasnet0_5 | PyTorch | Torchvision |
| mnasnet1_0 | PyTorch | Torchvision |
| mobilenet_v2 | PyTorch | Torchvision |
| mobilenet_v3_large | PyTorch | Torchvision |
| mobilenet_v3_small | PyTorch | Torchvision |
| regnet_x_16gf | PyTorch | Torchvision |
| regnet_x_1_6gf | PyTorch | Torchvision |
| regnet_x_32gf | PyTorch | Torchvision |
| regnet_x_3_2gf | PyTorch | Torchvision |
| regnet_x_400mf | PyTorch | Torchvision |
| regnet_x_800mf | PyTorch | Torchvision |
| regnet_x_8gf | PyTorch | Torchvision |
| regnet_y_16gf | PyTorch | Torchvision |
| regnet_y_1_6gf | PyTorch | Torchvision |
| regnet_y_32gf | PyTorch | Torchvision |
| regnet_y_3_2gf | PyTorch | Torchvision |
| regnet_y_400mf | PyTorch | Torchvision |
| regnet_y_800mf | PyTorch | Torchvision |
| regnet_y_8gf | PyTorch | Torchvision |
| resnet101 | PyTorch | Torchvision |
| resnet152 | PyTorch | Torchvision |
| resnet18 | PyTorch | Torchvision |
| resnet34 | PyTorch | Torchvision |
| resnet50 | PyTorch | Torchvision |
| resnext101_32x8d | PyTorch | Torchvision |
| resnext50_32x4d | PyTorch | Torchvision |
| shufflenet_v2_x0_5 | PyTorch | Torchvision |
| shufflenet_v2_x1_0 | PyTorch | Torchvision |
| vgg11 | PyTorch | Torchvision |
| vgg11_bn | PyTorch | Torchvision |
| vgg13 | PyTorch | Torchvision |
| vgg13_bn | PyTorch | Torchvision |
| vgg16 | PyTorch | Torchvision |
| vgg16_bn | PyTorch | Torchvision |
| vgg19 | PyTorch | Torchvision |
| vgg19_bn | PyTorch | Torchvision |
| vit_b_16 | PyTorch | Torchvision |
| vit_b_32 | PyTorch | Torchvision |
| vit_l_16 | PyTorch | Torchvision |
| vit_l_32 | PyTorch | Torchvision |
| wide_resnet101_2 | PyTorch | Torchvision |
| wide_resnet50_2 | PyTorch | Torchvision |
Text Generation
| Model name | Framework | Model Hub |
|---|---|---|
| bloom-560m | PyTorch | Hugging Face |
| bloomz-3b | PyTorch | Hugging Face |
| bloomz-560m | PyTorch | Hugging Face |
| distilgpt2 | PyTorch | Hugging Face |
| gpt-j-6b | PyTorch | Hugging Face |
| gpt2 | PyTorch | Hugging Face |