Here is a list of models that can be imported from src:
| Model |
Resolution |
PE |
Name |
Pretrained Weights |
Config |
| ViT-2/4 |
32x32 |
Learnable |
vit_2_4_32 |
- |
- |
| Sinusoidal |
vit_2_4_32_sine |
- |
- |
| ViT-4/4 |
32x32 |
Learnable |
vit_4_4_32 |
- |
- |
| Sinusoidal |
vit_4_4_32_sine |
- |
- |
| ViT-6/4 |
32x32 |
Learnable |
vit_6_4_32 |
- |
- |
| Sinusoidal |
vit_6_4_32_sine |
- |
- |
| ViT-7/4 |
32x32 |
Learnable |
vit_7_4_32 |
- |
- |
| Sinusoidal |
vit_7_4_32_sine |
- |
- |
| Model |
Resolution |
PE |
Name |
Pretrained Weights |
Config |
| CVT-2/4 |
32x32 |
Learnable |
cvt_2_4_32 |
- |
- |
| Sinusoidal |
cvt_2_4_32_sine |
- |
- |
| CVT-4/4 |
32x32 |
Learnable |
cvt_4_4_32 |
- |
- |
| Sinusoidal |
cvt_4_4_32_sine |
- |
- |
| CVT-6/4 |
32x32 |
Learnable |
cvt_6_4_32 |
- |
- |
| Sinusoidal |
cvt_6_4_32_sine |
- |
- |
| CVT-7/4 |
32x32 |
Learnable |
cvt_7_4_32 |
- |
- |
| Sinusoidal |
cvt_7_4_32_sine |
- |
- |
| Model |
Resolution |
PE |
Name |
Pretrained Weights |
Config |
| CCT-2/3x2 |
32x32 |
Learnable |
cct_2_3x2_32 |
- |
- |
| Sinusoidal |
cct_2_3x2_32_sine |
- |
- |
| CCT-4/3x2 |
32x32 |
Learnable |
cct_4_3x2_32 |
- |
- |
| Sinusoidal |
cct_4_3x2_32_sine |
- |
- |
| CCT-6/3x1 |
32x32 |
Learnable |
cct_6_3x1_32 |
- |
- |
| Sinusoidal |
cct_6_3x1_32_sine |
- |
- |
| CCT-6/3x2 |
32x32 |
Learnable |
cct_6_3x2_32 |
- |
- |
| Sinusoidal |
cct_6_3x2_32_sine |
- |
- |
| CCT-7/3x2 |
32x32 |
Learnable |
cct_7_3x2_32 |
- |
- |
| Sinusoidal |
cct_7_3x2_32_sine |
- |
- |
| CCT-7/3x1 |
32x32 |
Learnable |
cct_7_3x1_32 |
CIFAR-10/300 Epochs |
pretrained/cct_7-3x1_cifar10_300epochs.yml |
| Sinusoidal |
cct_7_3x1_32_sine |
CIFAR-10/5000 Epochs |
pretrained/cct_7-3x1_cifar10_5000epochs.yml |
| Learnable |
cct_7_3x1_32_c100 |
CIFAR-100/300 Epochs |
pretrained/cct_7-3x1_cifar100_300epochs.yml |
| Sinusoidal |
cct_7_3x1_32_sine_c100 |
CIFAR-100/5000 Epochs |
pretrained/cct_7-3x1_cifar100_5000epochs.yml |
| CCT-7/7x2 |
224x224 |
Learnable |
cct_7_7x2_224 |
- |
- |
| Sinusoidal |
cct_7_7x2_224_sine |
Flowers-102/300 Epochs |
pretrained/cct_7-7x2_flowers102.yml |
| CCT-14/7x2 |
224x224 |
Learnable |
cct_14_7x2_224 |
ImageNet-1k/300 Epochs |
pretrained/cct_14-7x2_imagenet.yml |
| 384x384 |
cct_14_7x2_384 |
ImageNet-1k/Finetuned/30 Epochs |
finetuned/cct_14-7x2_imagenet384.yml |
| 384x384 |
cct_14_7x2_384_fl |
Flowers102/Finetuned/300 Epochs |
finetuned/cct_14-7x2_flowers102.yml |