├── LICENSE ├── README.md ├── distributed_train.sh ├── metaformer_baselines.py ├── misc ├── in21k_to_1k_class_map.json └── process_21k_ckpt_before_1k_finetuning.py ├── scripts ├── caformer │ ├── finetune_caformer_b36_384.sh │ ├── finetune_caformer_m36_384.sh │ ├── finetune_caformer_s18_384.sh │ ├── finetune_caformer_s36_384.sh │ ├── train_caformer_b36.sh │ ├── train_caformer_m36.sh │ ├── train_caformer_s18.sh │ └── train_caformer_s36.sh ├── convformer │ ├── finetune_convformer_b36_384.sh │ ├── finetune_convformer_m36_384.sh │ ├── finetune_convformer_s18_384.sh │ ├── finetune_convformer_s36_384.sh │ ├── train_convformer_b36.sh │ ├── train_convformer_m36.sh │ ├── train_convformer_s18.sh │ └── train_convformer_s36.sh ├── identityformer │ ├── train_identityformer_m36.sh │ ├── train_identityformer_m48.sh │ ├── train_identityformer_s12.sh │ ├── train_identityformer_s24.sh │ └── train_identityformer_s36.sh ├── poolformerv2 │ ├── train_poolformerv2_m36.sh │ ├── train_poolformerv2_m48.sh │ ├── train_poolformerv2_s12.sh │ ├── train_poolformerv2_s24.sh │ └── train_poolformerv2_s36.sh └── randformer │ ├── train_randformer_m36.sh │ ├── train_randformer_m48.sh │ ├── train_randformer_s12.sh │ ├── train_randformer_s24.sh │ └── train_randformer_s36.sh ├── train.py ├── utils.py └── validate.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/README.md -------------------------------------------------------------------------------- /distributed_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/distributed_train.sh -------------------------------------------------------------------------------- /metaformer_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/metaformer_baselines.py -------------------------------------------------------------------------------- /misc/in21k_to_1k_class_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/misc/in21k_to_1k_class_map.json -------------------------------------------------------------------------------- /misc/process_21k_ckpt_before_1k_finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/misc/process_21k_ckpt_before_1k_finetuning.py -------------------------------------------------------------------------------- /scripts/caformer/finetune_caformer_b36_384.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/caformer/finetune_caformer_b36_384.sh -------------------------------------------------------------------------------- /scripts/caformer/finetune_caformer_m36_384.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/caformer/finetune_caformer_m36_384.sh -------------------------------------------------------------------------------- /scripts/caformer/finetune_caformer_s18_384.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/caformer/finetune_caformer_s18_384.sh -------------------------------------------------------------------------------- /scripts/caformer/finetune_caformer_s36_384.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/caformer/finetune_caformer_s36_384.sh -------------------------------------------------------------------------------- /scripts/caformer/train_caformer_b36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/caformer/train_caformer_b36.sh -------------------------------------------------------------------------------- /scripts/caformer/train_caformer_m36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/caformer/train_caformer_m36.sh -------------------------------------------------------------------------------- /scripts/caformer/train_caformer_s18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/caformer/train_caformer_s18.sh -------------------------------------------------------------------------------- /scripts/caformer/train_caformer_s36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/caformer/train_caformer_s36.sh -------------------------------------------------------------------------------- /scripts/convformer/finetune_convformer_b36_384.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/convformer/finetune_convformer_b36_384.sh -------------------------------------------------------------------------------- /scripts/convformer/finetune_convformer_m36_384.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/convformer/finetune_convformer_m36_384.sh -------------------------------------------------------------------------------- /scripts/convformer/finetune_convformer_s18_384.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/convformer/finetune_convformer_s18_384.sh -------------------------------------------------------------------------------- /scripts/convformer/finetune_convformer_s36_384.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/convformer/finetune_convformer_s36_384.sh -------------------------------------------------------------------------------- /scripts/convformer/train_convformer_b36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/convformer/train_convformer_b36.sh -------------------------------------------------------------------------------- /scripts/convformer/train_convformer_m36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/convformer/train_convformer_m36.sh -------------------------------------------------------------------------------- /scripts/convformer/train_convformer_s18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/convformer/train_convformer_s18.sh -------------------------------------------------------------------------------- /scripts/convformer/train_convformer_s36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/convformer/train_convformer_s36.sh -------------------------------------------------------------------------------- /scripts/identityformer/train_identityformer_m36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/identityformer/train_identityformer_m36.sh -------------------------------------------------------------------------------- /scripts/identityformer/train_identityformer_m48.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/identityformer/train_identityformer_m48.sh -------------------------------------------------------------------------------- /scripts/identityformer/train_identityformer_s12.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/identityformer/train_identityformer_s12.sh -------------------------------------------------------------------------------- /scripts/identityformer/train_identityformer_s24.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/identityformer/train_identityformer_s24.sh -------------------------------------------------------------------------------- /scripts/identityformer/train_identityformer_s36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/identityformer/train_identityformer_s36.sh -------------------------------------------------------------------------------- /scripts/poolformerv2/train_poolformerv2_m36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/poolformerv2/train_poolformerv2_m36.sh -------------------------------------------------------------------------------- /scripts/poolformerv2/train_poolformerv2_m48.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/poolformerv2/train_poolformerv2_m48.sh -------------------------------------------------------------------------------- /scripts/poolformerv2/train_poolformerv2_s12.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/poolformerv2/train_poolformerv2_s12.sh -------------------------------------------------------------------------------- /scripts/poolformerv2/train_poolformerv2_s24.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/poolformerv2/train_poolformerv2_s24.sh -------------------------------------------------------------------------------- /scripts/poolformerv2/train_poolformerv2_s36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/poolformerv2/train_poolformerv2_s36.sh -------------------------------------------------------------------------------- /scripts/randformer/train_randformer_m36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/randformer/train_randformer_m36.sh -------------------------------------------------------------------------------- /scripts/randformer/train_randformer_m48.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/randformer/train_randformer_m48.sh -------------------------------------------------------------------------------- /scripts/randformer/train_randformer_s12.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/randformer/train_randformer_s12.sh -------------------------------------------------------------------------------- /scripts/randformer/train_randformer_s24.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/randformer/train_randformer_s24.sh -------------------------------------------------------------------------------- /scripts/randformer/train_randformer_s36.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/scripts/randformer/train_randformer_s36.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/utils.py -------------------------------------------------------------------------------- /validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sail-sg/metaformer/HEAD/validate.py --------------------------------------------------------------------------------