Intel® Transfer Learning Tool Supported Models¶
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 |