Skip to content

Latest commit

 

History

History
296 lines (289 loc) · 8.11 KB

Variants.md

File metadata and controls

296 lines (289 loc) · 8.11 KB
Raw

Full list of variants provided

Here is a list of models that can be imported from src:

ViT-Lite

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 - -

CVT

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 - -

CCT

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