├── .coveragerc ├── .github ├── CONTRIBUTING.md ├── ISSUE_TEMPLATE │ ├── ----.md │ ├── ---.md │ ├── ---bug.md │ ├── bug_report.md │ ├── config.yml │ ├── feature_request.md │ └── general-questions.md └── pull_request_template.md ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── .style.yapf ├── LICENSE ├── MANIFEST.in ├── README.md ├── benchmarks ├── classification │ ├── README.md │ ├── dist_train_ft_4gpu.sh │ ├── dist_train_ft_8gpu.sh │ ├── dist_train_linear.sh │ ├── dist_train_linear_1gpu.sh │ ├── dist_train_linear_2gpu.sh │ ├── dist_train_linear_4gpu.sh │ ├── dist_train_semi.sh │ ├── knn_imagenet │ │ ├── dist_test_knn_epoch.sh │ │ ├── dist_test_knn_pretrain.sh │ │ ├── knn_classifier.py │ │ ├── slurm_test_knn_epoch.sh │ │ ├── slurm_test_knn_pretrain.sh │ │ └── test_knn.py │ ├── srun_train_ft.sh │ ├── srun_train_linear.sh │ ├── srun_train_semi.sh │ └── svm_voc07 │ │ ├── dist_extract.sh │ │ ├── dist_test_svm_epoch.sh │ │ ├── dist_test_svm_pretrain.sh │ │ ├── extract.py │ │ ├── slurm_extract.sh │ │ ├── slurm_test_svm_epoch.sh │ │ ├── slurm_test_svm_pretrain.sh │ │ └── tools │ │ ├── aggregate_low_shot_svm_stats.py │ │ ├── eval_svm_full.sh │ │ ├── eval_svm_lowshot.sh │ │ ├── svm_helper.py │ │ ├── test_svm.py │ │ ├── test_svm_low_shot.py │ │ ├── train_svm_kfold.py │ │ ├── train_svm_kfold_parallel.py │ │ ├── train_svm_low_shot.py │ │ └── train_svm_low_shot_parallel.py ├── detectron2 │ ├── README.md │ ├── convert-pretrain-to-detectron2.py │ ├── run.sh │ ├── slurm_run.sh │ └── train_net.py ├── mmdetection │ ├── README.md │ ├── mim_dist_test.sh │ ├── mim_dist_train.sh │ ├── mim_dist_train_c4.sh │ ├── mim_dist_train_full.sh │ ├── mim_slurm_test.sh │ ├── mim_slurm_train.sh │ ├── mim_slurm_train_c4.sh │ └── res_layer_extra_norm.py └── mmsegmentation │ ├── README.md │ ├── mim_dist_test.sh │ ├── mim_dist_train.sh │ ├── mim_dist_train_full.sh │ ├── mim_slurm_test.sh │ └── mim_slurm_train.sh ├── configs ├── benchmarks │ ├── classification │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── cifar100_swin_ft_sz224_8xbs128.py │ │ │ │ ├── cifar100_sz224_4xbs64.py │ │ │ │ ├── cifar100_sz32_bs100.py │ │ │ │ ├── cifar10_sz224_4xbs64.py │ │ │ │ ├── cifar10_sz32_bs100.py │ │ │ │ ├── imagenet100_swin_ft_sz192_8xbs128.py │ │ │ │ ├── imagenet100_sz224_4xbs64.py │ │ │ │ ├── imagenet_color_sz224_4xbs64.py │ │ │ │ ├── imagenet_rsb_a2_ft_sz224_8xbs256.py │ │ │ │ ├── imagenet_rsb_a3_ft_sz160_4xbs512.py │ │ │ │ ├── imagenet_swin_ft_sz192_8xbs128.py │ │ │ │ ├── imagenet_swin_ft_sz224_8xbs128.py │ │ │ │ ├── imagenet_sz224_4xbs64.py │ │ │ │ ├── inatualist2018_sz224_4xbs64.py │ │ │ │ ├── place205_sz224_4xbs64.py │ │ │ │ ├── stl10_swin_ft_sz96_8xbs128.py │ │ │ │ └── stl10_sz96_4xbs64.py │ │ │ ├── default_runtime.py │ │ │ └── models │ │ │ │ ├── beit_base_p16.py │ │ │ │ ├── convnext_base.py │ │ │ │ ├── convnext_large.py │ │ │ │ ├── convnext_small.py │ │ │ │ ├── convnext_tiny.py │ │ │ │ ├── deit_small_p16.py │ │ │ │ ├── r18.py │ │ │ │ ├── r18_cifar.py │ │ │ │ ├── r50.py │ │ │ │ ├── r50_cifar.py │ │ │ │ ├── r50_multihead.py │ │ │ │ ├── r50_rsb_a2.py │ │ │ │ ├── r50_rsb_a3.py │ │ │ │ ├── r50_swin.py │ │ │ │ ├── swin-base.py │ │ │ │ ├── swin-tiny.py │ │ │ │ ├── vit_base_deit_p16.py │ │ │ │ ├── vit_base_p16.py │ │ │ │ ├── vit_base_p16_linear.py │ │ │ │ ├── vit_huge_p14.py │ │ │ │ ├── vit_large_p16.py │ │ │ │ ├── vit_small_deit_p16.py │ │ │ │ ├── vit_small_p16.py │ │ │ │ └── vit_small_p16_linear.py │ │ ├── cifar10 │ │ │ ├── r18_linear_sz224_4xb64_cos_ep100.py │ │ │ ├── r18_linear_sz224_4xb64_step_ep100.py │ │ │ ├── r18_sz32_bs100_cos_ep200.py │ │ │ └── r50_linear_sz224_4xb64_cos_ep100.py │ │ ├── cifar100 │ │ │ ├── deit_small_p16_linear_sz224_4xb128_cos_ep100.py │ │ │ ├── r18_linear_sz224_4xb64_cos_ep100.py │ │ │ ├── r18_linear_sz224_4xb64_step_ep100.py │ │ │ ├── r18_sz32_bs100_cos_ep200.py │ │ │ └── r50_linear_sz224_4xb64_cos_ep100.py │ │ ├── imagenet │ │ │ ├── beit_base_p16_ft_sz224_8xb128_cos_ep100.py │ │ │ ├── convnext_b_spark_ft_sz224_8xb256_cos_fp16_ep300.py │ │ │ ├── convnext_s_spark_ft_sz224_8xb256_cos_fp16_ep300.py │ │ │ ├── convnext_t_rsb_a3_ft_sz160_8xb256_cos_fp16_ep100.py │ │ │ ├── convnext_t_spark_ft_sz224_8xb256_cos_fp16_ep400.py │ │ │ ├── deit_small_p16_swin_ft_sz224_4xb256_cos_ep100.py │ │ │ ├── imagenet_per_1 │ │ │ │ ├── r50_sz224_4xb64_head100_lr0_001_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head100_lr0_01_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head100_lr0_1_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head10_lr0_001_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head10_lr0_01_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head10_lr0_1_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head1_lr0_001_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head1_lr0_01_step_ep20.py │ │ │ │ └── r50_sz224_4xb64_head1_lr0_1_step_ep20.py │ │ │ ├── imagenet_per_10 │ │ │ │ ├── r50_sz224_4xb64_head100_lr0_001_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head100_lr0_01_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head100_lr0_1_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head10_lr0_001_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head10_lr0_01_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head10_lr0_1_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head1_lr0_001_step_ep20.py │ │ │ │ ├── r50_sz224_4xb64_head1_lr0_01_step_ep20.py │ │ │ │ └── r50_sz224_4xb64_head1_lr0_1_step_ep20.py │ │ │ ├── r101_rsb_a2_ft_sz224_8xb256_cos_fp16_ep300.py │ │ │ ├── r101_rsb_a3_ft_sz160_8xb256_cos_fp16_ep100.py │ │ │ ├── r152_rsb_a2_ft_sz224_8xb256_cos_fp16_ep300.py │ │ │ ├── r152_rsb_a3_ft_sz160_8xb256_cos_fp16_ep100.py │ │ │ ├── r18_linear_sz224_4xb64_step_ep100.py │ │ │ ├── r200_rsb_a2_ft_sz224_8xb256_cos_fp16_ep300.py │ │ │ ├── r200_rsb_a3_ft_sz160_8xb256_cos_fp16_ep100.py │ │ │ ├── r50_ft_sz224_4xb256_accu2_cos_ep100.py │ │ │ ├── r50_linear_sz224_4xb64_step_ep100.py │ │ │ ├── r50_mhead_sz224_4xb64_step_ep90.py │ │ │ ├── r50_rsb_a2_ft_sz224_8xb256_cos_fp16_ep300.py │ │ │ ├── r50_rsb_a3_ft_sz160_4xb512_cos_fp16_ep100.py │ │ │ ├── r50_swin_ft_sz192_4xb512_accu1_cos_ep100.py │ │ │ ├── r50_swin_ft_sz224_4xb256_accu2_cos_ep100.py │ │ │ ├── r50_swin_ft_sz224_8xb128_accu2_cos_ep100.py │ │ │ ├── swin_base_swin_ft_sz192_4xb128_accu4_cos_ep100.py │ │ │ ├── swin_base_swin_ft_sz224_4xb128_accu4_cos_ep100.py │ │ │ ├── swin_tiny_swin_ft_sz192_4xb128_accu4_cos_ep100.py │ │ │ ├── swin_tiny_swin_ft_sz224_4xb128_accu4_cos_ep100.py │ │ │ ├── vit_base_p16_swin_ft_cae_sz224_4xb128_accu2_cos_ep100.py │ │ │ ├── vit_base_p16_swin_ft_mae_sz224_8xb128_cos_ep100.py │ │ │ ├── vit_base_p16_swin_ft_simmim_sz224_4xb128_accu2_cos_ep100.py │ │ │ ├── vit_base_p16_swin_ft_simmim_sz224_8xb128_cos_ep100.py │ │ │ ├── vit_huge_p14_swin_ft_mae_sz224_8xb128_cos_ep50.py │ │ │ ├── vit_large_p16_swin_ft_mae_sz224_8xb128_cos_ep100.py │ │ │ ├── vit_large_p16_swin_ft_simmim_sz224_8xb64_accu2_cos_ep100.py │ │ │ ├── vit_large_p16_swin_ft_simmim_sz224_8xb64_accu2_cos_ep150.py │ │ │ ├── vit_small_p16_linear_sz224_8xb128_cos_ep90.py │ │ │ ├── vit_small_p16_linear_sz224_8xb128_step_ep90.py │ │ │ ├── vit_small_p16_swin_ft_simmim_sz224_8xb128_cos_ep100.py │ │ │ └── vit_small_p16_swin_ft_simmim_sz224_8xb128_cos_ep200.py │ │ ├── imagenet100 │ │ │ ├── deit_small_p16_swin_ft_sz224_4xb256_cos_ep100.py │ │ │ ├── r18_linear_sz224_4xb64_step_ep100.py │ │ │ ├── r50_ft_sz224_4xb256_accu2_cos_ep100.py │ │ │ ├── r50_linear_sz224_4xb64_step_ep100.py │ │ │ ├── r50_swin_ft_sz224_4xb256_accu2_cos_ep100.py │ │ │ ├── r50_swin_ft_sz224_8xb128_accu2_cos_ep100.py │ │ │ ├── swin_base_swin_ft_sz224_4xb128_accu4_cos_ep100.py │ │ │ ├── swin_tiny_swin_ft_sz224_4xb256_accu2_cos_ep100.py │ │ │ ├── vit_base_p16_swin_ft_mae_sz224_8xb128_cos_ep100.py │ │ │ └── vit_small_p16_linear_sz224_8xb128_cos_ep90.py │ │ ├── inaturalist2018 │ │ │ ├── r50_mhead_sz224_4xb64_step_ep84.py │ │ │ └── r50_sobel_mhead_sz224_4xb64_step_ep84.py │ │ ├── place205 │ │ │ ├── r50_mhead_sz224_4xb64_step_ep28.py │ │ │ └── r50_sobel_mhead_sz224_4xb64_step_ep28.py │ │ └── stl10 │ │ │ ├── deit_small_p16_linear_sz96_4xb128_cos_ep100.py │ │ │ ├── r50_linear_sz96_4xb64_cos_ep100.py │ │ │ └── r50_linear_sz96_4xb64_step_ep100.py │ ├── detectron2 │ │ ├── Base-Keypoint-RCNN-FPN.yaml │ │ ├── Base-RCNN-C4-BN.yaml │ │ ├── Base-RCNN-FPN.yaml │ │ ├── Base-RetinaNet.yaml │ │ ├── Cityscapes │ │ │ ├── mask_rcnn_R_50_FPN.yaml │ │ │ └── mask_rcnn_R_50_FPN_moco.yaml │ │ ├── coco_R_50_C4_1x.yaml │ │ ├── coco_R_50_C4_1x_moco.yaml │ │ ├── coco_R_50_C4_2x.yaml │ │ ├── coco_R_50_C4_2x_moco.yaml │ │ ├── coco_R_50_FPN_1x.yaml │ │ ├── coco_R_50_FPN_1x_moco.yaml │ │ ├── coco_R_50_FPN_2x.yaml │ │ ├── coco_R_50_FPN_2x_moco.yaml │ │ ├── coco_R_50_RetinaNet_1x.yaml │ │ ├── coco_R_50_RetinaNet_1x_moco.yaml │ │ ├── coco_R_50_RetinaNet_2x.yaml │ │ ├── coco_R_50_RetinaNet_2x_moco.yaml │ │ ├── keypoint_rcnn_R_50_FPN_2x.yaml │ │ ├── keypoint_rcnn_R_50_FPN_2x_moco.yaml │ │ ├── pascal_voc_R_50_C4_24k.yaml │ │ └── pascal_voc_R_50_C4_24k_moco.yaml │ ├── mmdetection │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── coco_instance.py │ │ │ │ └── voc0712.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ │ ├── faster_rcnn_r50_caffe_c4.py │ │ │ │ ├── faster_rcnn_r50_fpn.py │ │ │ │ ├── mask_rcnn_r50_caffe_c4.py │ │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ │ └── mask_rcnn_vit_fpn.py │ │ │ └── schedules │ │ │ │ ├── schedule_1x.py │ │ │ │ └── schedule_2x.py │ │ ├── coco │ │ │ ├── mask-rcnn_r50-c4_ms-1x_coco.py │ │ │ ├── mask-rcnn_r50-c4_ms-2x_coco.py │ │ │ ├── mask-rcnn_r50_fpn_ms-1x_coco.py │ │ │ ├── mask-rcnn_r50_fpn_ms-2x_coco.py │ │ │ └── mask-rcnn_vit_base_giou_4conv1f_1x_coco.py │ │ └── voc0712 │ │ │ └── faster-rcnn_r50-c4_ms-24k_voc0712.py │ └── mmsegmentation │ │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ ├── ade20k_640x640.py │ │ │ ├── cityscapes.py │ │ │ ├── cityscapes_769x769.py │ │ │ ├── pascal_voc12.py │ │ │ └── pascal_voc12_aug.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── fcn_r50-d8.py │ │ │ ├── upernet_beit.py │ │ │ ├── upernet_convnext.py │ │ │ ├── upernet_mae.py │ │ │ ├── upernet_r50.py │ │ │ └── upernet_swin.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ └── schedule_80k.py │ │ ├── ade20k │ │ ├── fcn_r50_d8_4x4_512x512_160k_ade20k.py │ │ ├── upernet_beit_base_8x2_512x512_160k_ade20k.py │ │ ├── upernet_beit_base_8x2_640x640_160k_ade20k.py │ │ └── upernet_mae_base_8x2_512x512_160k_ade20k.py │ │ ├── cityscapes │ │ └── fcn_r50-d8_769x769_40k_cityscapes.py │ │ └── voc12aug │ │ └── fcn_r50-d8_512x512_20k_voc12aug.py ├── classification │ ├── _base_ │ │ ├── datasets │ │ │ ├── aircrafts │ │ │ │ ├── sz224_bs16.py │ │ │ │ └── sz448_bs16.py │ │ │ ├── cifar10 │ │ │ │ └── sz32_bs100.py │ │ │ ├── cifar100 │ │ │ │ ├── sz224_corruption_bs100.py │ │ │ │ ├── sz224_randaug_bs100.py │ │ │ │ ├── sz224_swin_bs100.py │ │ │ │ ├── sz32_bs100.py │ │ │ │ ├── sz32_randaug_bs100.py │ │ │ │ └── sz32_swin_bs100.py │ │ │ ├── cub200 │ │ │ │ ├── sz224_bs16.py │ │ │ │ └── sz448_bs16.py │ │ │ ├── imagenet │ │ │ │ ├── autoaug_sz224_4xbs64.py │ │ │ │ ├── basic_sz224_4xbs64.py │ │ │ │ ├── basic_sz256_4xbs64.py │ │ │ │ ├── basic_sz299_4xbs64.py │ │ │ │ ├── basic_tta_sz224_4xbs64.py │ │ │ │ ├── convmixer_sz224_8xbs128.py │ │ │ │ ├── davit_sz224_8xbs128.py │ │ │ │ ├── deit3_ft_sz224_8xbs128.py │ │ │ │ ├── deit3_sz160_8xbs128.py │ │ │ │ ├── deit3_sz192_8xbs128.py │ │ │ │ ├── deit3_sz224_8xbs128.py │ │ │ │ ├── deit_adan_sz224_8xbs256.py │ │ │ │ ├── edgenext_sz256_8xbs128.py │ │ │ │ ├── lighting_sz224_4xbs64.py │ │ │ │ ├── mim_feat_sz224_4xbs256.py │ │ │ │ ├── mim_randaug_rgb_m_sz224_4xbs256.py │ │ │ │ ├── mim_rgb_m_sz224_4xbs256.py │ │ │ │ ├── mim_sz224_4xbs256.py │ │ │ │ ├── mobilenet_v3_sz224_8xbs128.py │ │ │ │ ├── mobilevit_sz256_8xbs128.py │ │ │ │ ├── moga_light_deit3_sz256_8xbs128.py │ │ │ │ ├── moga_light_sz224_8xbs128.py │ │ │ │ ├── moga_light_sz256_8xbs128.py │ │ │ │ ├── moga_sz224_8xbs128.py │ │ │ │ ├── poolformer_m_sz224_8xbs128.py │ │ │ │ ├── poolformer_s_sz224_8xbs128.py │ │ │ │ ├── rsb_a2_sz224_8xbs256.py │ │ │ │ ├── rsb_a3_sz160_4xbs512.py │ │ │ │ ├── rsb_a3_sz160_8xbs256.py │ │ │ │ ├── starnet_sz224_8xbs256.py │ │ │ │ ├── swin_sz224_4xbs256.py │ │ │ │ ├── swin_sz224_8xbs128.py │ │ │ │ ├── swin_sz384_8xbs64.py │ │ │ │ ├── t2t_vit_sz224_8xbs64.py │ │ │ │ └── van_sz224_8xbs128.py │ │ │ ├── imagenet100 │ │ │ │ ├── basic_sz224_4xbs256.py │ │ │ │ ├── mim_feat_sz224_4xbs256.py │ │ │ │ ├── mim_randaug_rgb_m_sz224_4xbs256.py │ │ │ │ ├── mim_rgb_m_sz224_4xbs256.py │ │ │ │ ├── mim_sz224_4xbs256.py │ │ │ │ └── swin_sz224_4xbs256.py │ │ │ ├── imagenet21k │ │ │ │ ├── basic_sz224_8xbs128.py │ │ │ │ └── swin_sz224_8xbs128.py │ │ │ ├── inaturalist2017 │ │ │ │ └── basic_sz224_4xbs64.py │ │ │ ├── inaturalist2018 │ │ │ │ └── basic_sz224_4xbs64.py │ │ │ ├── place205 │ │ │ │ └── basic_sz224_4xbs64.py │ │ │ └── tiny_imagenet │ │ │ │ ├── sz64_2xbs50.py │ │ │ │ └── sz64_bs100.py │ │ ├── default_runtime.py │ │ └── models │ │ │ ├── beit │ │ │ └── beit_base_p16.py │ │ │ ├── context_cluster │ │ │ ├── coc_medium.py │ │ │ ├── coc_small.py │ │ │ └── coc_tiny.py │ │ │ ├── convmixer │ │ │ ├── convmixer_1024_d20.py │ │ │ ├── convmixer_1536_d20.py │ │ │ └── convmixer_768_d32.py │ │ │ ├── convnext │ │ │ ├── convnext_base.py │ │ │ ├── convnext_large.py │ │ │ ├── convnext_small.py │ │ │ ├── convnext_tiny.py │ │ │ └── convnext_xlarge.py │ │ │ ├── convnext_v2 │ │ │ ├── convnext_v2_atto.py │ │ │ ├── convnext_v2_base.py │ │ │ ├── convnext_v2_femto.py │ │ │ ├── convnext_v2_huge.py │ │ │ ├── convnext_v2_large.py │ │ │ ├── convnext_v2_nano.py │ │ │ ├── convnext_v2_pico.py │ │ │ └── convnext_v2_tiny.py │ │ │ ├── davit │ │ │ ├── davit_base.py │ │ │ ├── davit_small.py │ │ │ └── davit_tiny.py │ │ │ ├── deit │ │ │ ├── deit_base_p16_sz224.py │ │ │ ├── deit_small_p16_sz224.py │ │ │ └── deit_tiny_p16_sz224.py │ │ │ ├── deit3 │ │ │ ├── deit3_base_p16_sz192.py │ │ │ ├── deit3_huge_p16_sz160.py │ │ │ ├── deit3_large_p16_sz192.py │ │ │ └── deit3_small_p16_sz224.py │ │ │ ├── densenet │ │ │ ├── densenet121.py │ │ │ ├── densenet161.py │ │ │ ├── densenet169.py │ │ │ └── densenet201.py │ │ │ ├── edgenext │ │ │ ├── edgenext-base.py │ │ │ ├── edgenext-small.py │ │ │ ├── edgenext-xsmall.py │ │ │ └── edgenext-xxsmall.py │ │ │ ├── efficientformer │ │ │ ├── efficientformer_l1.py │ │ │ ├── efficientformer_l3.py │ │ │ └── efficientformer_l7.py │ │ │ ├── efficientnet │ │ │ ├── efficientnet_b0.py │ │ │ ├── efficientnet_b1.py │ │ │ ├── efficientnet_b2.py │ │ │ ├── efficientnet_b3.py │ │ │ ├── efficientnet_b4.py │ │ │ ├── efficientnet_b5.py │ │ │ ├── efficientnet_b6.py │ │ │ ├── efficientnet_b7.py │ │ │ ├── efficientnet_b8.py │ │ │ ├── efficientnet_em.py │ │ │ └── efficientnet_es.py │ │ │ ├── efficientnet_v2 │ │ │ ├── efficientnet_v2_b0.py │ │ │ ├── efficientnet_v2_b1.py │ │ │ ├── efficientnet_v2_b2.py │ │ │ ├── efficientnet_v2_b3.py │ │ │ ├── efficientnet_v2_l.py │ │ │ ├── efficientnet_v2_m.py │ │ │ ├── efficientnet_v2_s.py │ │ │ └── efficientnet_v2_xl.py │ │ │ ├── hornet │ │ │ ├── hornet_base.py │ │ │ ├── hornet_large.py │ │ │ ├── hornet_small.py │ │ │ ├── hornet_small_gf.py │ │ │ ├── hornet_tiny.py │ │ │ └── hornet_tiny_gf.py │ │ │ ├── hrnet │ │ │ ├── hrnet_w18.py │ │ │ ├── hrnet_w30.py │ │ │ ├── hrnet_w32.py │ │ │ ├── hrnet_w40.py │ │ │ ├── hrnet_w44.py │ │ │ ├── hrnet_w48.py │ │ │ └── hrnet_w64.py │ │ │ ├── levit │ │ │ └── levit_256_p16.py │ │ │ ├── lit_v2 │ │ │ ├── lit_v2_base.py │ │ │ ├── lit_v2_medium.py │ │ │ └── lit_v2_small.py │ │ │ ├── metaformer │ │ │ ├── caformer_b36.py │ │ │ ├── caformer_m36.py │ │ │ ├── caformer_s18.py │ │ │ ├── caformer_s36.py │ │ │ ├── convformer_b36.py │ │ │ ├── convformer_m36.py │ │ │ ├── convformer_s18.py │ │ │ ├── convformer_s36.py │ │ │ ├── identityformer_m36.py │ │ │ ├── identityformer_s12.py │ │ │ ├── poolformerv2_m36.py │ │ │ ├── poolformerv2_s12.py │ │ │ ├── randformer_m36.py │ │ │ └── randformer_s12.py │ │ │ ├── mlp_mixer │ │ │ ├── mlp_mixer_base_p16.py │ │ │ ├── mlp_mixer_large_p16.py │ │ │ └── mlp_mixer_small_p16.py │ │ │ ├── mobilenet_v2 │ │ │ └── mobilenet_v2_1x.py │ │ │ ├── mobilenet_v3 │ │ │ ├── mobilenet_v3_large.py │ │ │ ├── mobilenet_v3_small.py │ │ │ ├── mobilenet_v3_small_050.py │ │ │ └── mobilenet_v3_small_075.py │ │ │ ├── mobileone │ │ │ ├── mobileone_s0.py │ │ │ ├── mobileone_s1.py │ │ │ ├── mobileone_s2.py │ │ │ ├── mobileone_s3.py │ │ │ └── mobileone_s4.py │ │ │ ├── mobilevit │ │ │ ├── mobilevit_s.py │ │ │ ├── mobilevit_xs.py │ │ │ └── mobilevit_xxs.py │ │ │ ├── moganet │ │ │ ├── moga_base.py │ │ │ ├── moga_large.py │ │ │ ├── moga_small.py │ │ │ ├── moga_tiny.py │ │ │ ├── moga_xlarge.py │ │ │ └── moga_xtiny.py │ │ │ ├── mvit │ │ │ ├── mvit_v2_base.py │ │ │ ├── mvit_v2_large.py │ │ │ ├── mvit_v2_small.py │ │ │ └── mvit_v2_tiny.py │ │ │ ├── poolformer │ │ │ ├── poolformer_m36.py │ │ │ ├── poolformer_m48.py │ │ │ ├── poolformer_s12.py │ │ │ ├── poolformer_s24.py │ │ │ └── poolformer_s36.py │ │ │ ├── pvt │ │ │ ├── pvt_large.py │ │ │ ├── pvt_medium.py │ │ │ ├── pvt_small.py │ │ │ └── pvt_tiny.py │ │ │ ├── regnet │ │ │ ├── regnetx_12gf.py │ │ │ ├── regnetx_1_6gf.py │ │ │ ├── regnetx_3_2gf.py │ │ │ ├── regnetx_400mf.py │ │ │ ├── regnetx_4_0gf.py │ │ │ ├── regnetx_6_4gf.py │ │ │ ├── regnetx_800mf.py │ │ │ └── regnetx_8_0gf.py │ │ │ ├── repmlp │ │ │ └── regmlp_base.py │ │ │ ├── repvgg │ │ │ ├── repvgg_A0.py │ │ │ ├── repvgg_B0.py │ │ │ └── repvgg_B3.py │ │ │ ├── res2net │ │ │ ├── res2net101_w26_s4.py │ │ │ ├── res2net50_w14_s8.py │ │ │ ├── res2net50_w26_s4.py │ │ │ ├── res2net50_w26_s6.py │ │ │ ├── res2net50_w26_s8.py │ │ │ └── res2net50_w48_s2.py │ │ │ ├── resnet │ │ │ ├── resnet101.py │ │ │ ├── resnet152.py │ │ │ ├── resnet18.py │ │ │ ├── resnet18_cifar.py │ │ │ ├── resnet18_rsb_a3.py │ │ │ ├── resnet34.py │ │ │ ├── resnet50.py │ │ │ ├── resnet50_mixup.py │ │ │ ├── resnet50_rsb_a3.py │ │ │ ├── resnet50_v1c.py │ │ │ ├── resnet50_v1d.py │ │ │ ├── resnext101_32x4d.py │ │ │ ├── resnext101_32x8d.py │ │ │ ├── resnext50_32x4d.py │ │ │ ├── seresnet50.py │ │ │ └── seresnext50_32x4d.py │ │ │ ├── rwkv │ │ │ ├── rwkv_base.py │ │ │ ├── rwkv_small.py │ │ │ └── rwkv_tiny.py │ │ │ ├── shufflenet_v1 │ │ │ └── shufflenet_v1_1x.py │ │ │ ├── shufflenet_v2 │ │ │ └── shufflenet_v2_1x.py │ │ │ ├── starnet │ │ │ ├── starnet_s1.py │ │ │ ├── starnet_s2.py │ │ │ ├── starnet_s3.py │ │ │ └── starnet_s4.py │ │ │ ├── swin_transformer │ │ │ ├── swin_base_sz224.py │ │ │ ├── swin_base_sz384.py │ │ │ ├── swin_large_sz224.py │ │ │ ├── swin_large_sz384.py │ │ │ ├── swin_small_sz224.py │ │ │ └── swin_tiny_sz224.py │ │ │ ├── t2t_vit │ │ │ ├── t2t_vit_tiny_14.py │ │ │ ├── t2t_vit_tiny_19.py │ │ │ └── t2t_vit_tiny_24.py │ │ │ ├── transnext │ │ │ ├── transnext_base.py │ │ │ ├── transnext_micro.py │ │ │ ├── transnext_small.py │ │ │ └── transnext_tiny.py │ │ │ ├── twins │ │ │ ├── twins_pcpvt_base.py │ │ │ └── twins_svt_base.py │ │ │ ├── uniformer │ │ │ ├── uniformer_base.py │ │ │ ├── uniformer_large.py │ │ │ ├── uniformer_small.py │ │ │ ├── uniformer_small_plus.py │ │ │ └── uniformer_tiny.py │ │ │ ├── unireplknet │ │ │ ├── unireplknet_a.py │ │ │ ├── unireplknet_f.py │ │ │ ├── unireplknet_n.py │ │ │ ├── unireplknet_p.py │ │ │ ├── unireplknet_s.py │ │ │ └── unireplknet_t.py │ │ │ ├── van │ │ │ ├── van_b4.py │ │ │ ├── van_b5.py │ │ │ ├── van_b6.py │ │ │ ├── van_base.py │ │ │ ├── van_large.py │ │ │ ├── van_small.py │ │ │ └── van_tiny.py │ │ │ ├── vanillanet │ │ │ ├── vanillanet_10.py │ │ │ ├── vanillanet_5.py │ │ │ ├── vanillanet_6.py │ │ │ ├── vanillanet_7.py │ │ │ ├── vanillanet_8.py │ │ │ └── vanillanet_9.py │ │ │ ├── vgg │ │ │ ├── vgg11.py │ │ │ ├── vgg11_bn.py │ │ │ ├── vgg13.py │ │ │ ├── vgg13_bn.py │ │ │ ├── vgg16.py │ │ │ ├── vgg16_bn.py │ │ │ ├── vgg19.py │ │ │ └── vgg19_bn.py │ │ │ └── vision_transformer │ │ │ ├── vit_base_p16_sz224.py │ │ │ ├── vit_large_p16_sz224.py │ │ │ └── vit_small_p16_sz224.py │ ├── aircrafts │ │ ├── README.md │ │ ├── adautomix │ │ │ └── basic │ │ │ │ ├── r18_adautomix_bilinear.py │ │ │ │ └── rx50_adautomix_bilinear.py │ │ ├── automix │ │ │ ├── r18_l2_a2_near_lam_cat_mb_mlr5e_4_1xb16_ep200.py │ │ │ └── rx50_l2_a2_near_lam_cat_mb_mlr5e_4_1xb16_ep200.py │ │ ├── mergemix │ │ │ └── vits │ │ │ │ ├── deit_small │ │ │ │ └── deit_s_mergemix_l12_a1_tm98_s_ep200.py │ │ │ │ └── vit_base │ │ │ │ └── vit_b_mergemix_l12_a1_tm98_s_ep200.py │ │ ├── mixups │ │ │ ├── auto_train_mixups.py │ │ │ └── basic │ │ │ │ ├── r18_alignmix_CE_none.py │ │ │ │ ├── r18_attentivemix_CE_none.py │ │ │ │ ├── r18_mixups_CE_none.py │ │ │ │ ├── r18_puzzlemix_CE_soft.py │ │ │ │ ├── r50_mixups_CE_none.py │ │ │ │ └── rx50_mixups_CE_none.py │ │ └── samix │ │ │ ├── r18_l2_a2_bili_val_dp01_mb_mlr5e_4_1xb16_ep200.py │ │ │ └── rx50_l2_a2_bili_val_dp01_mb_mlr5e_4_1xb16_ep200.py │ ├── cars │ │ ├── adautomix │ │ │ └── basic │ │ │ │ ├── r18_adautomix_bilinear.py │ │ │ │ └── rx50_adautomix_bilinear.py │ │ ├── automix │ │ │ └── basic │ │ │ │ ├── r18_automix_nearest.py │ │ │ │ └── rx50_automix_nearest.py │ │ ├── mergemix │ │ │ └── vits │ │ │ │ ├── deit_small │ │ │ │ └── deit_s_mergemix_l12_a1_tm98_s_ep200.py │ │ │ │ └── vit_base │ │ │ │ └── vit_b_mergemix_l12_a1_tm98_s_ep200.py │ │ └── mixups │ │ │ └── basic │ │ │ ├── r18_mixups_CE_none.py │ │ │ └── rx50_mixups_CE_none.py │ ├── cifar10 │ │ ├── README.md │ │ └── mixups │ │ │ ├── auto_train_mixups.py │ │ │ └── basic │ │ │ ├── r18 │ │ │ └── mix_modemixup │ │ │ │ └── r18_mixups_CE_none_alpha1_ep800.py │ │ │ ├── r18_alignmix_CE_none.py │ │ │ ├── r18_attentivemix_CE_none.py │ │ │ ├── r18_mixups_CE_none.py │ │ │ ├── r18_mixups_CE_none_multi_mode.py │ │ │ ├── r18_mixups_CE_sigm.py │ │ │ ├── r18_mixups_CE_soft.py │ │ │ ├── r18_puzzlemix_CE_soft.py │ │ │ ├── rx50_mixups_CE_none.py │ │ │ └── rx50_mixups_CE_soft.py │ ├── cifar100 │ │ ├── README.md │ │ ├── adautomix │ │ │ ├── auto_train_adautomix.py │ │ │ ├── basic │ │ │ │ ├── r18_l2_a1_bili_mlr5e_2.py │ │ │ │ └── rx50_l2_a1_bili_mlr5e_2.py │ │ │ └── vit │ │ │ │ ├── deit_s_l6_a1_near_sz224_mlr1e_4_bs100_ep200.py │ │ │ │ └── swin_t_l3_a1_near_sz224_mlr1e_4_bs100_ep200.py │ │ ├── advanced │ │ │ ├── convext_s_adamw_bs100_ep200.py │ │ │ ├── convext_t_adamw_bs100_ep200.py │ │ │ ├── deit_b_adamw_sz224_bs100_ep200.py │ │ │ ├── deit_s_adamw_sz224_bs100_ep200.py │ │ │ ├── densenet_121_adamw_bs100_ep200.py │ │ │ ├── densenet_169_adamw_bs100_ep200.py │ │ │ ├── densenet_201_adamw_bs100_ep200.py │ │ │ ├── mlpmixer_s_adamw_sz224_bs100_ep200.py │ │ │ ├── moganet_b_adamw_bs100_ep200.py │ │ │ ├── moganet_s_adamw_bs100_ep200.py │ │ │ ├── poolformer_m36_adamw_bs100_ep200.py │ │ │ ├── poolformer_m48_adamw_bs100_ep200.py │ │ │ ├── poolformer_s24_adamw_bs100_ep200.py │ │ │ ├── poolformer_s36_adamw_bs100_ep200.py │ │ │ ├── resnet_101_adamw_bs100_ep200.py │ │ │ ├── resnet_152_adamw_bs100_ep200.py │ │ │ ├── resnet_50_adamw_bs100_ep200.py │ │ │ ├── swin_b_adamw_sz224_bs100_ep200.py │ │ │ ├── swin_l_adamw_sz224_bs100_ep200.py │ │ │ ├── swin_s_adamw_sz224_bs100_ep200.py │ │ │ ├── swin_t_adamw_sz224_bs100_ep200.py │ │ │ ├── twin_pcpvt_b_adamw_sz224_bs100_ep200.py │ │ │ ├── twin_pcpvt_s_adamw_sz224_bs100_ep200.py │ │ │ ├── twin_svt_b_adamw_sz224_bs100_ep200.py │ │ │ └── twin_svt_s_adamw_sz224_bs100_ep200.py │ │ ├── automix │ │ │ ├── auto_train_cifar_automix.py │ │ │ ├── basic │ │ │ │ ├── r18 │ │ │ │ │ └── unsampling_modenearest │ │ │ │ │ │ └── r18_l2_a2_near_L1_01_mlr5e_2_ep800.py │ │ │ │ ├── r18_l2_a2_near_L1_01_mlr5e_2.py │ │ │ │ ├── rx50_l2_a2_near_L1_01_mlr5e_2.py │ │ │ │ └── wrn28_8_l1_a2_near_L1_01_mlr1e_3.py │ │ │ └── vits │ │ │ │ ├── convnext_t_l2_a2_near_val_dp01_lr2e_3_mlr1e_4_bs100_ep200.py │ │ │ │ ├── convnext_t_l2_a2_near_val_dp01_lr2e_3_mlr1e_4_bs100_ep600.py │ │ │ │ ├── deit_s_l6_a2_near_L1_01_sz224_mlr1e_4_bs100_ep200.py │ │ │ │ ├── deit_s_l6_a2_near_L1_01_sz224_mlr1e_4_bs100_ep600.py │ │ │ │ ├── swin_t_l3_a2_near_L1_01_sz224_mlr1e_4_bs100_ep200.py │ │ │ │ └── swin_t_l3_a2_near_bili_L1_01_sz224_mlr2e_4_bs100_ep600.py │ │ ├── mixups │ │ │ ├── auto_train_mixups.py │ │ │ ├── basic │ │ │ │ ├── r18 │ │ │ │ │ ├── mix_modealignmix │ │ │ │ │ │ └── r18_mixups_CE_none_alpha2_ep800.py │ │ │ │ │ ├── mix_modeattentivemix │ │ │ │ │ │ └── r18_attentivemix_CE_none_alpha2_ep800.py │ │ │ │ │ ├── mix_modemixup │ │ │ │ │ │ └── r18_mixups_CE_none_alpha1_ep800.py │ │ │ │ │ └── mix_modepuzzlemix │ │ │ │ │ │ └── r18_mixups_CE_soft_alpha1_ep800.py │ │ │ │ ├── r18_adautomix_bilinear.py │ │ │ │ ├── r18_alignmix_CE_none.py │ │ │ │ ├── r18_attentivemix_CE_none.py │ │ │ │ ├── r18_automix_near.py │ │ │ │ ├── r18_mixups_CE_none.py │ │ │ │ ├── r18_mixups_CE_none_multi_mode.py │ │ │ │ ├── r18_mixups_CE_sigm.py │ │ │ │ ├── r18_mixups_CE_soft.py │ │ │ │ ├── r18_puzzlemix_CE_soft.py │ │ │ │ ├── rx50_adautomix.py │ │ │ │ ├── rx50_automix_near.py │ │ │ │ ├── rx50_mixups_CE_none.py │ │ │ │ ├── rx50_mixups_CE_soft.py │ │ │ │ └── wrn28_8_mixups_CE_none.py │ │ │ ├── decouple │ │ │ │ ├── r18 │ │ │ │ │ ├── r18_cutmix_CE_soft_decouple_a0_2_ep800.py │ │ │ │ │ ├── r18_fmix_CE_soft_decouple_a0_2_ep800.py │ │ │ │ │ ├── r18_manifoldmix_CE_soft_decouple_a0_5_ep800.py │ │ │ │ │ ├── r18_mixup_CE_soft_decouple_a0_2_ep800.py │ │ │ │ │ ├── r18_resizemix_CE_soft_decouple_a0_2_ep800.py │ │ │ │ │ └── r18_saliencymix_CE_soft_decouple_a0_2_ep800.py │ │ │ │ ├── r18_mixups_CE_soft_decouple.py │ │ │ │ ├── rx50 │ │ │ │ │ ├── rx50_cutmix_CE_soft_decouple_a0_2_ep800.py │ │ │ │ │ ├── rx50_fmix_CE_soft_decouple_a0_2_ep800.py │ │ │ │ │ ├── rx50_manifoldmix_CE_soft_decouple_a2_ep800.py │ │ │ │ │ ├── rx50_mixup_CE_soft_decouple_a0_2_ep800.py │ │ │ │ │ ├── rx50_resizemix_CE_soft_decouple_a0_2_ep800.py │ │ │ │ │ └── rx50_saliencymix_CE_soft_decouple_a0_2_ep800.py │ │ │ │ └── rx50_mixups_CE_soft_decouple.py │ │ │ └── vits │ │ │ │ ├── convnext_t_mixups_bs100.py │ │ │ │ ├── convnext_tiny │ │ │ │ ├── convnext_t_alignmix_a2_bs100_ep200.py │ │ │ │ ├── convnext_t_attentivemix_a2_bs100_ep200.py │ │ │ │ ├── convnext_t_automix_l2_a2_near_bs100_ep200.py │ │ │ │ ├── convnext_t_cutmix_a2_bs100_ep200.py │ │ │ │ ├── convnext_t_deit_bs100_ep200.py │ │ │ │ ├── convnext_t_fmix_a1_bs100_ep200.py │ │ │ │ ├── convnext_t_gridmix_a1_bs100_ep200.py │ │ │ │ ├── convnext_t_manifoldmix_a2_bs100_ep200.py │ │ │ │ ├── convnext_t_mixup_a0_8_bs100_ep200.py │ │ │ │ ├── convnext_t_puzzlemix_a2_bs100_ep200.py │ │ │ │ ├── convnext_t_resizemix_lam01_08_bs100_ep200.py │ │ │ │ ├── convnext_t_saliencymix_a0_2_bs100_ep200.py │ │ │ │ ├── convnext_t_samix_l2_a2_near_bs100_ep200.py │ │ │ │ ├── convnext_t_smoothmix_a0_2_bs100_ep200.py │ │ │ │ └── convnext_t_vanilla_bs100_ep200.py │ │ │ │ ├── deit_b_mixups_sz224_bs100.py │ │ │ │ ├── deit_base │ │ │ │ ├── deit_base_ep200 │ │ │ │ │ ├── deit_b_adautomix_l6_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_attentivemix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_automix_l6_a2_near_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_cutmix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_deit_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_fmix_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_gridmix_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_mergemix_l9_a1_tm98_s_sz224_4bs25_ep200.py │ │ │ │ │ ├── deit_b_mixpro_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_mixup_a0_8_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_puzzlemix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_resizemix_lam01_08_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_saliencymix_a0_2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_smmix_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_smoothmix_a0_2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_b_tla_sz224_bs100_ep200_wo_pre_trained.py │ │ │ │ │ ├── deit_b_transmix_p0_8_sz224_bs100_ep200.py │ │ │ │ │ └── deit_b_vanilla_sz224_bs100_ep200.py │ │ │ │ └── deit_base_ep600 │ │ │ │ │ ├── deit_b_adautomix_l6_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_attentivemix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_automix_l6_a2_near_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_cutmix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_deit_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_fmix_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_gridmix_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_mergemix_l9_a1_tm98_s_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_mixpro_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_mixup_a0_8_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_puzzlemix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_resizemix_lam01_08_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_saliencymix_a0_2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_smmix_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_smoothmix_a0_2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_b_transmix_p0_8_sz224_bs100_ep600.py │ │ │ │ │ └── deit_b_vanilla_sz224_bs100_ep600.py │ │ │ │ ├── deit_s_cifar100_c_eval_sz224_bs100.py │ │ │ │ ├── deit_s_mixups_sz224_bs100.py │ │ │ │ ├── deit_s_mixups_sz224_bs100_randaug.py │ │ │ │ ├── deit_small │ │ │ │ ├── deit_s_adautomix_l6_a1_near_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_adautomix_l6_a1_near_sz224_bs100_ep600.py │ │ │ │ ├── deit_s_attentivemix_a2_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_automix_l6_a2_near_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_cutmix_a2_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_deit_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_fmix_a1_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_gridmix_a1_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_mergemix_l9_a1_tm98_s_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_mixpro_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_mixup_a0_8_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_puzzlemix_a2_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_resizemix_lam01_08_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_saliencymix_a0_2_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_samix_l6_a2_near_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_smmix_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_smoothmix_a0_2_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_tla_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_tla_sz224_bs100_ep200_wo_pre_trained.py │ │ │ │ ├── deit_s_tokenmix_sz224_bs100_ep200.py │ │ │ │ ├── deit_s_transmix_p0_8_sz224_bs100_ep200.py │ │ │ │ └── deit_s_vanilla_sz224_bs100_ep200.py │ │ │ │ ├── deit_t_mixups_sz224_bs100.py │ │ │ │ ├── deit_tiny │ │ │ │ ├── deit_tiny_ep200 │ │ │ │ │ ├── deit_t_adautomix_l6_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_attentivemix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_automix_l6_a2_near_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_cutmix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_deit_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_fmix_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_gridmix_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_mergemix_l9_a1_tm98_s_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_mixpro_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_mixup_a0_8_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_puzzlemix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_resizemix_lam01_08_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_saliencymix_a0_2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_samix_l6_a2_near_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_smmix_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_smmix_sz224_bs100_ep200_offical.py │ │ │ │ │ ├── deit_t_smoothmix_a0_2_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_tla_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_tla_sz224_bs100_ep200_wo_pre_trained.py │ │ │ │ │ ├── deit_t_tokenmix_sz224_bs100_ep200.py │ │ │ │ │ ├── deit_t_transmix_p0_8_sz224_bs100_ep200.py │ │ │ │ │ └── deit_t_vanilla_sz224_bs100_ep200.py │ │ │ │ └── deit_tiny_ep600 │ │ │ │ │ ├── deit_t_adautomix_l6_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_attentivemix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_automix_l6_a2_near_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_cutmix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_deit_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_fmix_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_gridmix_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_mergemix_l9_a1_tm98_s_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_mixpro_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_mixup_a0_8_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_puzzlemix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_resizemix_lam01_08_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_saliencymix_a0_2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_samix_l6_a2_near_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_smmix_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_smmix_sz224_bs100_ep600_offical.py │ │ │ │ │ ├── deit_t_smoothmix_a0_2_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_tla_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_tla_sz224_bs100_ep600_wo_pre_trained.py │ │ │ │ │ ├── deit_t_tokenmix_sz224_bs100_ep600.py │ │ │ │ │ ├── deit_t_transmix_p0_8_sz224_bs100_ep600.py │ │ │ │ │ └── deit_t_vanilla_sz224_bs100_ep600.py │ │ │ │ ├── swin_t_cifar100_c_eval_sz224_bs100.py │ │ │ │ ├── swin_t_mixups_sz224_bs100.py │ │ │ │ ├── swin_tiny │ │ │ │ ├── swin_t_alignmix_a2_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_attentivemix_a2_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_automix_l3_a2_near_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_cutmix_a2_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_deit_sz224_bs100_ep600.py │ │ │ │ ├── swin_t_fmix_a1_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_gridmix_a1_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_mixpro_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_mixpro_sz224_bs100_ep600.py │ │ │ │ ├── swin_t_mixup_a0_8_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_puzzlemix_a2_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_resizemix_lam01_08_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_saliencymix_a0_2_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_samix_l3_a2_near_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_smmix_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_smmix_sz224_bs100_ep600.py │ │ │ │ ├── swin_t_smoothmix_a0_2_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_transmix_sz224_bs100_ep200.py │ │ │ │ ├── swin_t_transmix_sz224_bs100_ep600.py │ │ │ │ └── swin_t_vanilla_sz224_bs100_ep200.py │ │ │ │ ├── vit-large │ │ │ │ └── vit_large_ep200 │ │ │ │ │ ├── vit_l_adautomix_l4_a2_near_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_attentivemix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_automix_l4_a2_near_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_cutmix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_deit_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_fmix_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_gridmix_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_mergemix_18th_tome98_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_mixpro_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_mixup_a0_8_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_puzzlemix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_resizemix_lam01_08_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_saliencymix_a0_2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_smmix_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_smoothmix_a0_2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_l_transmix_p0_8_sz224_bs100_ep200.py │ │ │ │ │ └── vit_l_vanilla_sz224_bs100_ep200.py │ │ │ │ ├── vit-small │ │ │ │ ├── vit_small_ep200 │ │ │ │ │ ├── vit_s_adautomix_l4_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_attentivemix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_automix_l4_a2_near_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_cutmix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_deit_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_fmix_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_gridmix_a1_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_mergemix_l6_a1_tm98_s_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_mixpro_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_mixup_a0_8_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_puzzlemix_a2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_resizemix_lam01_08_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_saliencymix_a0_2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_smmix_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_smoothmix_a0_2_sz224_bs100_ep200.py │ │ │ │ │ ├── vit_s_transmix_p0_8_sz224_bs100_ep200.py │ │ │ │ │ └── vit_s_vanilla_sz224_bs100_ep200.py │ │ │ │ └── vit_small_ep600 │ │ │ │ │ ├── vit_s_adautomix_l4_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_attentivemix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_automix_l4_a2_near_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_cutmix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_deit_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_fmix_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_gridmix_a1_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_mergemix_l6_a1_tm98_s_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_mixpro_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_mixup_a0_8_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_puzzlemix_a2_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_resizemix_lam01_08_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_saliencymix_a0_2_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_smmix_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_smoothmix_a0_2_sz224_bs100_ep600.py │ │ │ │ │ ├── vit_s_transmix_p0_8_sz224_bs100_ep600.py │ │ │ │ │ └── vit_s_vanilla_sz224_bs100_ep600.py │ │ │ │ ├── vit_l_mixups_sz224_bs100.py │ │ │ │ └── vit_s_mixups_sz224_bs100.py │ │ ├── optimizers │ │ │ └── basic │ │ │ │ ├── r50_CE_none_bs100.py │ │ │ │ ├── r50_CE_none_bs100_adabelief_ep200.py │ │ │ │ ├── r50_CE_none_bs100_adafactor_ep200.py │ │ │ │ ├── r50_CE_none_bs100_adam_ep200.py │ │ │ │ ├── r50_CE_none_bs100_adamp_ep200.py │ │ │ │ ├── r50_CE_none_bs100_adamw_ep200.py │ │ │ │ ├── r50_CE_none_bs100_adan_ep200.py │ │ │ │ ├── r50_CE_none_bs100_lamb_ep200.py │ │ │ │ ├── r50_CE_none_bs100_sam_adam_ep200.py │ │ │ │ ├── r50_CE_none_bs100_sam_adamw_ep200.py │ │ │ │ ├── r50_CE_none_bs100_sam_sgd_ep200.py │ │ │ │ └── r50_CE_none_bs100_sgd_ep200.py │ │ ├── samix │ │ │ ├── auto_train_cifar_samix.py │ │ │ ├── basic │ │ │ │ ├── r18 │ │ │ │ │ └── unsampling_modebilinear │ │ │ │ │ │ └── r18_l2_a2_bili_val_dp0_mul_x_cat_L1_var_01_mlr5e_2_lam_mul_k0_25_mask_adjust0_25_ep800.py │ │ │ │ ├── r18_l2_a2_bili_val_dp0_mul_x_cat_L1_var_01_mlr5e_2.py │ │ │ │ ├── rx50_l2_a2_bili_val_dp0_mul_x_cat_L1_var_01_mlr5e_2.py │ │ │ │ └── wrn28_8_l1_a2_bili_val_dp0_mul_x_cat_L1_var_01_mlr1e_3.py │ │ │ └── vits │ │ │ │ ├── convnext_t_l2_a2_bili_val_dp01_lr2e_3_mlr1e_4_bs100_ep200.py │ │ │ │ ├── convnext_t_l2_a2_bili_val_dp01_lr2e_3_mlr1e_4_bs100_ep600.py │ │ │ │ ├── deit_s_l6_a2_near_val_dp01_sz224_mlr1e_4_bs100_ep200.py │ │ │ │ ├── deit_s_l6_a2_near_val_dp01_sz224_mlr1e_4_bs100_ep600.py │ │ │ │ ├── swin_t_l3_a2_near_bili_val_dp01_sz224_mlr2e_4_bs100_ep600.py │ │ │ │ └── swin_t_l3_a2_near_val_dp01_sz224_mlr1e_4_bs100_ep200.py │ │ └── wa │ │ │ ├── convext_s_CE_adamw_bs100.py │ │ │ ├── convext_t_CE_adamw_bs100.py │ │ │ ├── deit_s_CE_adamw_sz224_bs100.py │ │ │ ├── densenet121_CE_bs100.py │ │ │ ├── ema │ │ │ ├── convext_s_CE_adamw_bs100_ema_ep200.py │ │ │ ├── convext_t_CE_adamw_bs100_ema_ep200.py │ │ │ ├── deit_s_CE_adamw_sz224_bs100_ema_ep200.py │ │ │ ├── densenet121_CE_bs100_ema_ep200.py │ │ │ ├── mlpmixer_s_CE_adamw_sz224_bs100_ema_ep200.py │ │ │ ├── resnet18_CE_bs100_ema_ep200.py │ │ │ ├── resnet50_CE_bs100_ema_ep200.py │ │ │ ├── resnext50_CE_bs100_ema_ep200.py │ │ │ ├── swin_t_CE_adamw_sz224_bs100_ema_ep200.py │ │ │ └── wrn28_10_CE_bs100_ema_ep200.py │ │ │ ├── mlpmixer_s_CE_adamw_sz224_bs100.py │ │ │ ├── resnet101_CE_bs100.py │ │ │ ├── resnet18_CE_bs100.py │ │ │ ├── resnext50_CE_bs100.py │ │ │ ├── swin_t_CE_adamw_sz224_bs100.py │ │ │ └── wrn28_10_CE_bs100.py │ ├── cub200 │ │ ├── README.md │ │ ├── adautomix │ │ │ ├── r18_adautomix_bilinear.py │ │ │ ├── r50_adautomix_bilinear.py │ │ │ └── rx50_adautomix_bilinear.py │ │ ├── automix │ │ │ ├── r18_l2_a2_near_lam_cat_mb_mlr5e_4_1xb16_ep200.py │ │ │ └── rx50_l2_a2_near_lam_cat_mb_mlr5e_4_1xb16_ep200.py │ │ ├── mergemix │ │ │ └── vits │ │ │ │ ├── deit_small │ │ │ │ └── deit_s_mergemix_l9_a1_tm147_s_ep200.py │ │ │ │ └── vit_base │ │ │ │ └── vit_b_mergemix_l9_a1_tm98_s_ep200.py │ │ ├── mixups │ │ │ ├── auto_train_mixups.py │ │ │ └── basic │ │ │ │ ├── r18_alignmix_CE_none.py │ │ │ │ ├── r18_attentivemix_CE_none.py │ │ │ │ ├── r18_mixups_CE_none.py │ │ │ │ ├── r18_puzzlemix_CE_soft.py │ │ │ │ ├── r50_mixups_CE_none.py │ │ │ │ └── rx50_mixups_CE_none.py │ │ └── samix │ │ │ ├── r18_l2_a2_bili_val_dp01_mb_mlr5e_4_1xb16_ep200.py │ │ │ └── rx50_l2_a2_bili_val_dp01_mb_mlr5e_4_1xb16_ep200.py │ ├── imagenet │ │ ├── adautomix │ │ │ ├── README.md │ │ │ └── basic │ │ │ │ ├── r18_adautomix_bilinear.py │ │ │ │ ├── r34_adautomix_bilinear.py │ │ │ │ └── r50_adautomix_bilinear.py │ │ ├── alexnet │ │ │ ├── README.md │ │ │ └── alexnet_4xb64_cos_ep100.py │ │ ├── automix │ │ │ ├── README.md │ │ │ ├── auto_train_in_automix.py │ │ │ ├── basic │ │ │ │ ├── r101_l2_a2_near_lam_cat_mb_mlr1e_3_bb_mlr0.py │ │ │ │ ├── r18_l2_a2_near_lam_cat_mb_mlr1e_3_bb_mlr0.py │ │ │ │ ├── r34_l2_a2_near_lam_cat_mb_mlr1e_3_bb_mlr0.py │ │ │ │ ├── r50 │ │ │ │ │ └── unsampling_modenearest │ │ │ │ │ │ ├── r50_l2_a2_near_lam_cat_mb_mlr1e_3_m09_bb_mlr0_ep300.py │ │ │ │ │ │ ├── r50_l2_a2_near_lam_cat_mb_mlr1e_3_m0_bb_mlr0_ep100.py │ │ │ │ │ │ └── r50_l2_a2_near_lam_cat_mb_mlr1e_3_m0_bb_mlr0_fp16_ep100.py │ │ │ │ ├── r50_l2_a2_bili_lam_cat_mb_mlr1e_3_bb_mlr0.py │ │ │ │ ├── r50_l2_a2_near_lam_cat_mb_mlr1e_3_bb_mlr0.py │ │ │ │ └── rx101_l2_a2_near_lam_cat_mb_mlr1e_3_bb_mlr0.py │ │ │ ├── convnext │ │ │ │ └── convnext_t_l2_a2_near_lam_cat_switch0_8_8x128_accu4_ep300.py │ │ │ ├── deit │ │ │ │ ├── deit_s_l6_a2_near_lam_cat_switch0_8_8x128_ep300.py │ │ │ │ └── deit_s_l6_a2_near_lam_cat_switch0_8_att_ln_8x128_fp16_ep300.py │ │ │ ├── pvt │ │ │ │ ├── pvt_t_l2_a2_near_lam_cat_swch0_8_8x128_fp16_ep300.py │ │ │ │ └── pvt_t_l2_a2_near_lam_cat_swch0_8_attn_ln_8x128_fp16_ep300.py │ │ │ └── swin │ │ │ │ ├── swin_t_l2_a2_near_lam_cat_switch0_8_8x128_ep300.py │ │ │ │ └── swin_t_l2_a2_near_lam_cat_switch0_8_att_ln_8x128_fp16_ep300.py │ │ ├── beit │ │ │ ├── README.md │ │ │ └── beit_base_8xb128_fp16_ep300.py │ │ ├── context_cluster │ │ │ ├── README.md │ │ │ ├── coc_medium_8xb256_ep300.py │ │ │ ├── coc_small_8xb256_ep300.py │ │ │ ├── coc_tiny_8xb256_ep300.py │ │ │ └── coc_tiny_plain_8xb256_ep300.py │ │ ├── convmixer │ │ │ ├── README.md │ │ │ ├── convmixer_1024_d20_8xb60_fp16_ep300.py │ │ │ ├── convmixer_1535_d20_8xb60_fp16_ep300.py │ │ │ ├── convmixer_768_d32_8xb80_fp16_ep300.py │ │ │ └── convmixer_768_d32_rsb_a3_sz160_8xb256_ep100.py │ │ ├── convnext │ │ │ ├── README.md │ │ │ ├── convnext_base_8xb128_accu4_fp16_ep300.py │ │ │ ├── convnext_base_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── convnext_large_8xb64_accu8_fp16_ep300.py │ │ │ ├── convnext_large_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── convnext_small_8xb128_accu4_fp16_ep300.py │ │ │ ├── convnext_small_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── convnext_tiny_8xb256_accu2_fp16_ep300.py │ │ │ ├── convnext_tiny_rsb_a3_sz160_8xb256_ep100.py │ │ │ └── convnext_xlarge_8xb64_accu8_fp16_ep300.py │ │ ├── convnext_v2 │ │ │ ├── README.md │ │ │ ├── convnext_v2_atto_8xb128_fp16_ep600.py │ │ │ ├── convnext_v2_base_8xb128_fp16_ep100.py │ │ │ ├── convnext_v2_femto_8xb128_fp16_ep600.py │ │ │ ├── convnext_v2_huge_8xb64_accu2_fp16_ep100.py │ │ │ ├── convnext_v2_large_8xb128_fp16_ep100.py │ │ │ ├── convnext_v2_nano_8xb128_fp16_ep600.py │ │ │ ├── convnext_v2_pico_8xb128_fp16_ep600.py │ │ │ └── convnext_v2_tiny_8xb128_fp16_ep300.py │ │ ├── cspnet │ │ │ ├── README.md │ │ │ ├── cspdarknet50_4xb64_ep100.py │ │ │ ├── cspresnet50_4xb64_ep100.py │ │ │ └── cspresnext50_4xb64_ep100.py │ │ ├── davit │ │ │ ├── README.md │ │ │ ├── davit_base_sz224_4xb256_ep300.py │ │ │ ├── davit_small_sz224_4xb256_ep300.py │ │ │ └── davit_tiny_sz224_4xb256_ep300.py │ │ ├── deit │ │ │ ├── README.md │ │ │ ├── deit_base_8xb128_ep300.py │ │ │ ├── deit_base_adan_8xb256_fp16_ep150.py │ │ │ ├── deit_base_adan_8xb256_fp16_ep300.py │ │ │ ├── deit_base_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── deit_small_8xb128_ep300.py │ │ │ ├── deit_small_adan_8xb256_fp16_ep150.py │ │ │ ├── deit_small_adan_8xb256_fp16_ep300.py │ │ │ ├── deit_small_adan_I_8xb256_fp16_ep150.py │ │ │ ├── deit_small_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── deit_tiny_8xb128_ep300.py │ │ │ └── deit_tiny_rsb_a3_sz160_8xb256_ep100.py │ │ ├── deit3 │ │ │ ├── README.md │ │ │ ├── deit3_base_sz192_8xb128_accu2_ep800.py │ │ │ ├── deit3_base_sz224_ft_4xb128_ep20.py │ │ │ ├── deit3_huge_sz160_8xb64_accu4_ep800.py │ │ │ ├── deit3_huge_sz224_ft_8xb64_ep20.py │ │ │ ├── deit3_large_sz192_8xb128_accu2_ep800.py │ │ │ ├── deit3_large_sz224_ft_8xb64_ep20.py │ │ │ └── deit3_small_sz224_8xb256_ep800.py │ │ ├── densenet │ │ │ ├── README.md │ │ │ ├── densenet121_4xb256_ep90.py │ │ │ ├── densenet161_4xb256_ep90.py │ │ │ ├── densenet169_4xb256_ep90.py │ │ │ └── densenet201_4xb256_ep90.py │ │ ├── edgenext │ │ │ ├── README.md │ │ │ ├── edgenext_base_sz256_8xb256_ep300.py │ │ │ ├── edgenext_small_sz256_8xb256_ep300.py │ │ │ ├── edgenext_xsmall_sz256_8xb256_ep300.py │ │ │ └── edgenext_xxsmall_sz256_8xb256_ep300.py │ │ ├── efficientformer │ │ │ ├── README.md │ │ │ ├── efficientformer_l1_8xb256_ep300.py │ │ │ ├── efficientformer_l3_8xb256_ep300.py │ │ │ └── efficientformer_l7_8xb256_ep300.py │ │ ├── efficientnet │ │ │ ├── README.md │ │ │ ├── efficientnet_b0_4xb64.py │ │ │ ├── efficientnet_b0_4xb64_05norm.py │ │ │ ├── efficientnet_b1_4xb64.py │ │ │ ├── efficientnet_b1_4xb64_05norm.py │ │ │ ├── efficientnet_b2_4xb64.py │ │ │ ├── efficientnet_b2_4xb64_05norm.py │ │ │ ├── efficientnet_b3_4xb64.py │ │ │ ├── efficientnet_b3_4xb64_05norm.py │ │ │ ├── efficientnet_b4_4xb64.py │ │ │ ├── efficientnet_b4_4xb64_05norm.py │ │ │ ├── efficientnet_b5_4xb64.py │ │ │ ├── efficientnet_b5_4xb64_05norm.py │ │ │ ├── efficientnet_b6_4xb64.py │ │ │ ├── efficientnet_b6_4xb64_05norm.py │ │ │ ├── efficientnet_b7_4xb64.py │ │ │ ├── efficientnet_b7_4xb64_05norm.py │ │ │ ├── efficientnet_b8_4xb64.py │ │ │ ├── efficientnet_b8_4xb64_05norm.py │ │ │ ├── efficientnet_em_4xb64_05norm.py │ │ │ └── efficientnet_es_4xb64_05norm.py │ │ ├── efficientnet_v2 │ │ │ ├── README.md │ │ │ ├── efficientnet_b0_4xb64.py │ │ │ ├── efficientnet_b1_4xb64.py │ │ │ ├── efficientnet_b2_4xb64.py │ │ │ ├── efficientnet_b3_4xb64.py │ │ │ ├── efficientnet_l_8xb32_sz480.py │ │ │ ├── efficientnet_m_8xb32_sz480.py │ │ │ ├── efficientnet_s_4xb32_sz384.py │ │ │ └── efficientnet_xl_8xb32_sz512.py │ │ ├── hornet │ │ │ ├── README.md │ │ │ ├── hornet_base_8xb64_ep300.py │ │ │ ├── hornet_base_gf_8xb64_ep300.py │ │ │ ├── hornet_small_8xb64_ep300.py │ │ │ ├── hornet_small_gf_8xb64_ep300.py │ │ │ ├── hornet_small_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── hornet_tiny_8xb128_ep300.py │ │ │ ├── hornet_tiny_gf_8xb128_ep300.py │ │ │ └── hornet_tiny_rsb_a3_sz160_8xb256_ep100.py │ │ ├── hrnet │ │ │ ├── README.md │ │ │ ├── hrnet_w18_4xb64_ep100.py │ │ │ ├── hrnet_w30_4xb64_ep100.py │ │ │ ├── hrnet_w32_4xb64_ep100.py │ │ │ ├── hrnet_w40_4xb64_ep100.py │ │ │ ├── hrnet_w44_4xb64_ep100.py │ │ │ ├── hrnet_w48_4xb64_ep100.py │ │ │ └── hrnet_w64_4xb64_ep100.py │ │ ├── inception_v3 │ │ │ ├── README.md │ │ │ ├── inception_v3_4xb64_cos_ep100.py │ │ │ └── inception_v3_aux_4xb64_cos_ep100.py │ │ ├── levit │ │ │ ├── README.md │ │ │ ├── deploy │ │ │ │ ├── levit_128_p16_8xb256_ep300.py │ │ │ │ ├── levit_128s_p16_8xb256_ep300.py │ │ │ │ ├── levit_192_p16_8xb256_ep300.py │ │ │ │ ├── levit_256_p16_8xb256_ep300.py │ │ │ │ └── levit_384_p16_8xb256_ep300.py │ │ │ ├── levit_128_p16_8xb256_ep300.py │ │ │ ├── levit_128s_p16_8xb256_ep300.py │ │ │ ├── levit_192_p16_8xb256_ep300.py │ │ │ ├── levit_256_p16_8xb256_ep300.py │ │ │ └── levit_384_p16_8xb256_ep300.py │ │ ├── lit_v2 │ │ │ ├── README.md │ │ │ ├── lit_v2_base_8xb128_cos_fp16_ep300.py │ │ │ ├── lit_v2_base_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── lit_v2_medium_8xb128_cos_fp16_ep300.py │ │ │ ├── lit_v2_medium_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── lit_v2_small_8xb128_cos_fp16_ep300.py │ │ │ └── lit_v2_small_rsb_a3_sz160_8xb256_ep100.py │ │ ├── mergemix │ │ │ └── deit │ │ │ │ └── deit_small │ │ │ │ └── deit_small_merge49_l9_s0_a1_4x256_ep300.py │ │ ├── metaformer │ │ │ ├── README.md │ │ │ ├── caformer_b36_adamw_8xb128_ac4_ep300.py │ │ │ ├── caformer_m36_adamw_8xb128_ac4_ep300.py │ │ │ ├── caformer_s18_adamw_8xb128_ac4_ep300.py │ │ │ ├── caformer_s36_adamw_8xb128_ac4_ep300.py │ │ │ ├── convformer_b36_8xb128_ac4_ep300.py │ │ │ ├── convformer_m36_8xb128_ac4_ep300.py │ │ │ ├── convformer_s18_8xb128_ac4_ep300.py │ │ │ ├── convformer_s36_8xb128_ac4_ep300.py │ │ │ ├── identityformer_m36_8xb128_ac4_ep300.py │ │ │ ├── identityformer_m48_8xb128_ac4_ep300.py │ │ │ ├── identityformer_s12_8xb128_ac4_ep300.py │ │ │ ├── identityformer_s24_8xb128_ac4_ep300.py │ │ │ ├── identityformer_s36_8xb128_ac4_ep300.py │ │ │ ├── poolformerv2_m36_8xb128_ac4_ep300.py │ │ │ ├── poolformerv2_m48_8xb128_ac4_ep300.py │ │ │ ├── poolformerv2_s12_8xb128_ac4_ep300.py │ │ │ ├── poolformerv2_s24_8xb128_ac4_ep300.py │ │ │ ├── poolformerv2_s36_8xb128_ac4_ep300.py │ │ │ ├── randformer_m36_8xb128_ac4_ep300.py │ │ │ ├── randformer_m48_8xb128_ac4_ep300.py │ │ │ ├── randformer_s12_8xb128_ac4_ep300.py │ │ │ ├── randformer_s24_8xb128_ac4_ep300.py │ │ │ └── randformer_s36_8xb128_ac4_ep300.py │ │ ├── mixups │ │ │ ├── README.md │ │ │ ├── auto_train_mixups.py │ │ │ ├── basic │ │ │ │ ├── r101_mixups_CE_none_4xb64.py │ │ │ │ ├── r18 │ │ │ │ │ └── mix_modemixup │ │ │ │ │ │ └── r18_mixups_CE_none_4xb64_alpha0_2_ep100.py │ │ │ │ ├── r18_mixups_CE_none_4xb64.py │ │ │ │ ├── r34_mixups_CE_none_4xb64.py │ │ │ │ ├── r50 │ │ │ │ │ ├── mix_modealignmix │ │ │ │ │ │ └── r50_mixups_CE_soft_4xb64_alpha2_ep100.py │ │ │ │ │ ├── mix_modeattentivemix │ │ │ │ │ │ └── r50_attentivemix_CE_none_4xb64_alpha2_ep100.py │ │ │ │ │ ├── mix_modepuzzlemix │ │ │ │ │ │ └── r50_mixups_CE_soft_4xb64_alpha1_ep100.py │ │ │ │ │ └── mix_modevanilla │ │ │ │ │ │ └── r50_mixups_CE_none_4xb64_ep100.py │ │ │ │ ├── r50_alignmix_CE_soft_4xb64.py │ │ │ │ ├── r50_attentivemix_CE_none_4xb64.py │ │ │ │ ├── r50_mixups_CE_none_4xb64.py │ │ │ │ ├── r50_mixups_CE_none_multi_mode_4xb64.py │ │ │ │ ├── r50_mixups_CE_sigm_4xb64.py │ │ │ │ ├── r50_mixups_CE_soft_4xb64.py │ │ │ │ ├── r50_mixups_CE_soft_multi_mode_4xb64.py │ │ │ │ ├── r50_puzzlemix_CE_soft_4xb64.py │ │ │ │ └── rx101_mixups_CE_none_4xb64.py │ │ │ ├── convnext │ │ │ │ ├── convnext_tiny_smooth_mix0_8_cut1_0_8xb256_accu2_ema_fp16.py │ │ │ │ └── convnext_tiny_smooth_mix_8xb256_accu2_ema_fp16.py │ │ │ ├── decouple │ │ │ │ ├── r18_mixups_CE_soft_decouple_4xb64.py │ │ │ │ └── r50_mixups_CE_soft_decouple_4xb64.py │ │ │ ├── deit │ │ │ │ ├── deit_base │ │ │ │ │ ├── deit_base_smooth_attentivemix_a2_8xb128_ep300.py │ │ │ │ │ ├── deit_base_smooth_cutmix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_base_smooth_fmix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_base_smooth_mixup_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_base_smooth_puzzlemix_a1_4xb256_ep300.py │ │ │ │ │ ├── deit_base_smooth_resizemix_lam01_08_4xb256_ep300.py │ │ │ │ │ ├── deit_base_smooth_saliencymix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_base_smooth_transmix_8xb128_ep300.py │ │ │ │ │ └── deit_base_smooth_vanilla_4xb256_ep300.py │ │ │ │ ├── deit_base_smooth_deit_8xb128_ep300.py │ │ │ │ ├── deit_base_smooth_mix_8xb128.py │ │ │ │ ├── deit_small │ │ │ │ │ ├── deit_small_smooth_attentivemix_a2_4xb256_ep300.py │ │ │ │ │ ├── deit_small_smooth_cutmix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_small_smooth_fmix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_small_smooth_mixup_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_small_smooth_puzzlemix_a1_4xb256_ep300.py │ │ │ │ │ ├── deit_small_smooth_resizemix_lam01_08_4xb256_ep300.py │ │ │ │ │ ├── deit_small_smooth_saliencymix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_small_smooth_transmix_4xb256_ep300.py │ │ │ │ │ ├── deit_small_smooth_vanilla_4xb256_ep300.py │ │ │ │ │ └── mix_modemixup │ │ │ │ │ │ └── deit_small_smooth_mix_8x128_alpha0_2_fp16_ep300.py │ │ │ │ ├── deit_small_smooth_deit_8xb128_ep300.py │ │ │ │ ├── deit_small_smooth_mix_4xb256_fp16.py │ │ │ │ ├── deit_small_smooth_mix_8xb128.py │ │ │ │ ├── deit_tiny │ │ │ │ │ ├── deit_tiny_smooth_attentivemix_a2_4xb256_ep300.py │ │ │ │ │ ├── deit_tiny_smooth_cutmix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_tiny_smooth_fmix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_tiny_smooth_mixup_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_tiny_smooth_puzzlemix_a1_4xb256_ep300.py │ │ │ │ │ ├── deit_tiny_smooth_resizemix_lam01_08_4xb256_ep300.py │ │ │ │ │ ├── deit_tiny_smooth_saliencymix_a0_2_4xb256_ep300.py │ │ │ │ │ ├── deit_tiny_smooth_transmix_4xb256_ep300.py │ │ │ │ │ ├── deit_tiny_smooth_vanilla_4xb256_ep300.py │ │ │ │ │ └── mix_modemixup │ │ │ │ │ │ └── deit_tiny_smooth_mix_4xb256_alpha0_2_fp16_ep300.py │ │ │ │ ├── deit_tiny_smooth_deit_4xb256_ep300.py │ │ │ │ └── deit_tiny_smooth_mix_4xb256.py │ │ │ ├── moganet │ │ │ │ ├── moganet_tiny_smooth_deit_4xb256.py │ │ │ │ └── moganet_tiny_smooth_mix_4xb256.py │ │ │ ├── pvt │ │ │ │ ├── pvt_medium_smooth_mix_8xb128.py │ │ │ │ ├── pvt_small_smooth_mix_4xb256.py │ │ │ │ ├── pvt_tiny_smooth_deit_4xb256.py │ │ │ │ └── pvt_tiny_smooth_mix_4xb256.py │ │ │ ├── rsb_a2 │ │ │ │ ├── deit_small_rsb_a2_CE_sigm_mix0_1_cut1_0_4xb256_accu2_fp16.py │ │ │ │ ├── efficientnet_b0_rsb_a2_CE_sigm_mix0_1_cut1_0_4xb256_accu2_fp16.py │ │ │ │ ├── mobilenet_v2_1x_rsb_a2_CE_sigm_mix0_1_cut1_0_4xb256_accu2_fp16.py │ │ │ │ └── r50_rsb_a2_CE_sigm_mix0_1_cut1_0_4xb256_accu2_fp16.py │ │ │ ├── rsb_a3 │ │ │ │ ├── efficientnet_b0_rsb_a3_CE_sigm_mix0_1_cut1_0_4xb512_fp16.py │ │ │ │ ├── mobilenet_v2_1x_rsb_a3_CE_sigm_mix0_1_cut1_0_4xb512_fp16.py │ │ │ │ ├── r18_rsb_a3_CE_sigm_mix0_1_cut1_0_4xb512_fp16.py │ │ │ │ ├── r50_rsb_a3_CE_none_4xb512_fp16.py │ │ │ │ ├── r50_rsb_a3_CE_sigm_4xb512_fp16.py │ │ │ │ ├── r50_rsb_a3_CE_sigm_mix0_1_cut1_0_4xb512_fp16.py │ │ │ │ ├── r50_rsb_a3_CE_sigm_multi_4xb512_fp16.py │ │ │ │ └── r50_rsb_a3_CE_soft_4xb512_fp16.py │ │ │ └── swin │ │ │ │ ├── swin_tiny_smooth_mix0_8_cut1_0_4xb256_fp16.py │ │ │ │ ├── swin_tiny_smooth_mix_4xb256_fp16.py │ │ │ │ └── swin_tiny_timm_smooth_mix0_8_cut1_0_4xb256.py │ │ ├── mlp_mixer │ │ │ ├── README.md │ │ │ ├── mlp_mixer_base_p16_8xb128_fp16_ep300.py │ │ │ ├── mlp_mixer_large_p16_8xb64_accu2_fp16_ep300.py │ │ │ └── mlp_mixer_small_p16_8xb128_fp16_ep300.py │ │ ├── mobilenet_v2 │ │ │ ├── README.md │ │ │ └── mobilenet_v2_1x_4xb64_ep300.py │ │ ├── mobilenet_v3 │ │ │ ├── README.md │ │ │ ├── mobilenet_v3_large_8xb128_ep600.py │ │ │ ├── mobilenet_v3_small_050_8xb128_ep600.py │ │ │ ├── mobilenet_v3_small_075_8xb128_ep600.py │ │ │ └── mobilenet_v3_small_8xb128_ep600.py │ │ ├── mobileone │ │ │ ├── README.md │ │ │ ├── deploy │ │ │ │ ├── mobileone_s0_4xb64_ep300.py │ │ │ │ ├── mobileone_s1_4xb64_ep300.py │ │ │ │ ├── mobileone_s2_4xb64_ep300.py │ │ │ │ ├── mobileone_s3_4xb64_ep300.py │ │ │ │ └── mobileone_s4_4xb64_ep300.py │ │ │ ├── mobileone_s0_4xb64_ep300.py │ │ │ ├── mobileone_s1_4xb64_ep300.py │ │ │ ├── mobileone_s2_4xb64_ep300.py │ │ │ ├── mobileone_s3_4xb64_ep300.py │ │ │ └── mobileone_s4_4xb64_ep300.py │ │ ├── mobilevit │ │ │ ├── README.md │ │ │ ├── mobilevit_small_8xb128_ep300.py │ │ │ ├── mobilevit_xsmall_8xb128_ep300.py │ │ │ └── mobilevit_xxsmall_8xb128_ep300.py │ │ ├── moganet │ │ │ ├── README.md │ │ │ ├── moga_base_ema_sz224_8xb128_ep300.py │ │ │ ├── moga_base_rsb_a3_sz160_8xb128_accu2_ep100.py │ │ │ ├── moga_base_sz224_8xb128_ep300.py │ │ │ ├── moga_large_ema_sz224_8xb64_accu2_ep300.py │ │ │ ├── moga_large_rsb_a3_sz160_8xb128_accu2_ep100.py │ │ │ ├── moga_large_sz224_8xb64_accu2_ep300.py │ │ │ ├── moga_small_ema_sz224_8xb128_ep300.py │ │ │ ├── moga_small_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── moga_small_sz224_8xb128_ep300.py │ │ │ ├── moga_tiny_deit3_sz256_lr2e_3_8xb128_fp16_ep300.py │ │ │ ├── moga_tiny_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── moga_tiny_sz224_8xb128_fp16_ep300.py │ │ │ ├── moga_tiny_sz256_8xb128_fp16_ep300.py │ │ │ ├── moga_xlarge_ema_sz224_8xb32_accu2_ep300.py │ │ │ ├── moga_xtiny_deit3_sz256_lr2e_3_8xb128_fp16_ep300.py │ │ │ ├── moga_xtiny_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── moga_xtiny_sz224_8xb128_fp16_ep300.py │ │ │ └── moga_xtiny_sz256_8xb128_fp16_ep300.py │ │ ├── mvit │ │ │ ├── README.md │ │ │ ├── mvit_v2_base_8xb256_ep300.py │ │ │ ├── mvit_v2_large_8xb256_ep300.py │ │ │ ├── mvit_v2_small_8xb256_ep300.py │ │ │ └── mvit_v2_tiny_8xb256_ep300.py │ │ ├── poolformer │ │ │ ├── README.md │ │ │ ├── poolformer_m36_8xb128_fp16_ep300.py │ │ │ ├── poolformer_m36_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── poolformer_m48_8xb128_fp16_ep300.py │ │ │ ├── poolformer_m48_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── poolformer_s12_8xb128_fp16_ep300.py │ │ │ ├── poolformer_s12_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── poolformer_s24_8xb128_fp16_ep300.py │ │ │ ├── poolformer_s24_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── poolformer_s36_8xb128_fp16_ep300.py │ │ │ └── poolformer_s36_rsb_a3_sz160_8xb256_ep100.py │ │ ├── pvt │ │ │ ├── README.md │ │ │ ├── pvt_large_8xb128_ep300.py │ │ │ ├── pvt_medium_8xb128_ep300.py │ │ │ ├── pvt_small_8xb128_ep300.py │ │ │ ├── pvt_small_rsb_a3_sz160_8xb256_ep300.py │ │ │ ├── pvt_tiny_8xb128_ep300.py │ │ │ └── pvt_tiny_rsb_a3_sz160_8xb256_ep300.py │ │ ├── regnet │ │ │ ├── README.md │ │ │ ├── regnetx_12gf_8xb64_ep100.py │ │ │ ├── regnetx_1_6gf_8xb128_ep100.py │ │ │ ├── regnetx_3_2gf_8xb64_ep100.py │ │ │ ├── regnetx_400mf_8xb128_ep100.py │ │ │ ├── regnetx_4_0gf_8xb64_ep100.py │ │ │ ├── regnetx_6_4gf_8xb64_ep100.py │ │ │ ├── regnetx_800mf_8xb128_ep100.py │ │ │ └── regnetx_8_0gf_8xb64_ep100.py │ │ ├── repmlp │ │ │ ├── README.md │ │ │ ├── repmlp_base_8xb64_ep300.py │ │ │ ├── repmlp_base_deploy_8xb64_ep300.py │ │ │ ├── repmlp_base_deploy_sz256_8xb64_ep300.py │ │ │ └── repmlp_base_sz256_8xb64_ep300.py │ │ ├── repvgg │ │ │ ├── README.md │ │ │ ├── deploy │ │ │ │ ├── repvgg_A0_deploy_4xb64_ep120.py │ │ │ │ ├── repvgg_A1_deploy_4xb64_ep120.py │ │ │ │ ├── repvgg_A2_deploy_4xb64_ep120.py │ │ │ │ ├── repvgg_B0_deploy_4xb64_ep120.py │ │ │ │ ├── repvgg_B1_deploy_4xb64_ep120.py │ │ │ │ ├── repvgg_B1g2_deploy_4xb64_ep120.py │ │ │ │ ├── repvgg_B1g4_deploy_4xb64_ep120.py │ │ │ │ ├── repvgg_B2_deploy_4xb64_ep120.py │ │ │ │ ├── repvgg_B2g4_deploy_autoaug_mixup_4xb64_ep200.py │ │ │ │ ├── repvgg_B3_deploy_autoaug_mixup_4xb64_ep200.py │ │ │ │ ├── repvgg_B3g4_deploy_autoaug_mixup_4xb64_ep200.py │ │ │ │ └── repvgg_D2se_deploy_autoaug_mixup_4xb64_ep200.py │ │ │ ├── repvgg_A0_4xb64_ep120.py │ │ │ ├── repvgg_A1_4xb64_ep120.py │ │ │ ├── repvgg_A2_4xb64_ep120.py │ │ │ ├── repvgg_A2_deit_8xb128_fp16_ep300.py │ │ │ ├── repvgg_B0_4xb64_ep120.py │ │ │ ├── repvgg_B0_deit_8xb128_fp16_ep300.py │ │ │ ├── repvgg_B1_4xb64_ep120.py │ │ │ ├── repvgg_B1g2_4xb64_ep120.py │ │ │ ├── repvgg_B1g4_4xb64_ep120.py │ │ │ ├── repvgg_B2_4xb64_ep120.py │ │ │ ├── repvgg_B2g4_autoaug_mixup_4xb64_ep200.py │ │ │ ├── repvgg_B3_autoaug_mixup_4xb64_ep200.py │ │ │ ├── repvgg_B3g4_autoaug_mixup_4xb64_ep200.py │ │ │ └── repvgg_D2se_autoaug_mixup_4xb64_ep200.py │ │ ├── res2net │ │ │ ├── README.md │ │ │ ├── res2net101_w26_s4_4xb64_ep100.py │ │ │ ├── res2net50_w14_s8_4xb64_ep100.py │ │ │ └── res2net50_w26_s8_4xb64_ep100.py │ │ ├── resnet │ │ │ ├── README.md │ │ │ ├── resnet101_4xb64_step_ep100.py │ │ │ ├── resnet101_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── resnet152_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── resnet154_4xb64_step_ep100.py │ │ │ ├── resnet18_4xb64_cos_ep100.py │ │ │ ├── resnet18_4xb64_step_ep100.py │ │ │ ├── resnet18_rsb_a3_sz160_4xb512_ep100.py │ │ │ ├── resnet34_4xb64_step_ep100.py │ │ │ ├── resnet34_rsb_a3_sz160_4xb512_ep100.py │ │ │ ├── resnet50_4xb64_cos_ep100.py │ │ │ ├── resnet50_4xb64_step_ep100.py │ │ │ ├── resnet50_in21k_semantic_sz224_8xb256_ep90.py │ │ │ ├── resnet50_rsb_a1_sz224_8xb256_ep600.py │ │ │ ├── resnet50_rsb_a2_sz224_8xb256_ep300.py │ │ │ ├── resnet50_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── resnetv1c50_4xb64_step_ep100.py │ │ │ ├── resnetv1d50_4xb64_step_ep100.py │ │ │ ├── resnext101_32x4d_4xb64_step_ep100.py │ │ │ ├── resnext154_32x4d_4xb64_step_ep100.py │ │ │ └── resnext50_32x4d_4xb64_step_ep100.py │ │ ├── rwkv │ │ │ ├── README.md │ │ │ ├── mha_pro_tiny_8xb128_ep300.py │ │ │ ├── mha_rotary_tiny_8xb128_ep300.py │ │ │ ├── mha_shift_tiny_8xb128_ep300.py │ │ │ ├── rwkv_base_8xb128_ep300.py │ │ │ ├── rwkv_small_8xb128_ep300.py │ │ │ └── rwkv_tiny_8xb128_ep300.py │ │ ├── samix │ │ │ ├── README.md │ │ │ ├── auto_train_in_samix.py │ │ │ └── basic │ │ │ │ ├── r101_l2_a2_bili_val_dp01_mul_mb_mlr1e_3_bb_mlr0_4xb64.py │ │ │ │ ├── r18_l2_a2_bili_val_dp01_mul_mb_mlr1e_3_bb_mlr0_4xb64.py │ │ │ │ ├── r34_l2_a2_bili_val_dp01_mul_mb_mlr1e_3_bb_mlr0_4xb64.py │ │ │ │ ├── r50 │ │ │ │ └── unsampling_modebilinear │ │ │ │ │ ├── r50_l2_a2_bili_val_dp01_mul_mb_mlr1e_3_m09_bb_mlr0_4xb64_ep300.py │ │ │ │ │ ├── r50_l2_a2_bili_val_dp01_mul_mb_mlr1e_3_m0_bb_mlr0_4xb64_ep100.py │ │ │ │ │ └── r50_l2_a2_bili_val_dp01_mul_mb_mlr1e_3_m0_bb_mlr0_4xb64_fp16_ep100.py │ │ │ │ ├── r50_l2_a2_bili_val_dp01_mul_mb_mlr1e_3_bb_mlr0_4xb64.py │ │ │ │ └── rx101_l2_a2_bili_val_dp01_mul_mb_mlr1e_3_bb_mlr0_4xb64.py │ │ ├── shufflenet_v1 │ │ │ ├── README.md │ │ │ └── shufflenet_v1_1x_8xb128_ep300.py │ │ ├── shufflenet_v2 │ │ │ ├── README.md │ │ │ └── shufflenet_v2_1x_8xb128_ep300.py │ │ ├── starnet │ │ │ ├── README.md │ │ │ ├── starnet_s1_8xb256_ep300.py │ │ │ ├── starnet_s2_8xb256_ep300.py │ │ │ ├── starnet_s3_8xb256_ep300.py │ │ │ └── starnet_s4_8xb256_ep300.py │ │ ├── swin_transformer │ │ │ ├── README.md │ │ │ ├── swin_base_8xb128_fp16_ep300.py │ │ │ ├── swin_base_rsb_a3_sz160_8xb256_ep300.py │ │ │ ├── swin_large_8xb64_accu2_fp16_ep300.py │ │ │ ├── swin_small_8xb128_fp16_ep300.py │ │ │ ├── swin_small_rsb_a3_sz160_8xb256_ep300.py │ │ │ ├── swin_tiny_8xb128_fp16_ep300.py │ │ │ └── swin_tiny_rsb_a3_sz160_8xb256_ep300.py │ │ ├── t2t_vit │ │ │ ├── README.md │ │ │ ├── t2t_vit_tiny_14_ema_8xb64_ep310.py │ │ │ ├── t2t_vit_tiny_19_ema_8xb64_ep310.py │ │ │ └── t2t_vit_tiny_24_ema_8xb64_ep310.py │ │ ├── transnext │ │ │ ├── README.md │ │ │ ├── transnext_base_8xb128_ep300.py │ │ │ ├── transnext_micro_8xb128_ep300.py │ │ │ ├── transnext_small_8xb128_ep300.py │ │ │ └── transnext_tiny_8xb128_ep300.py │ │ ├── twins │ │ │ ├── README.md │ │ │ ├── twins_pcpvt_base_8xb128_ep300.py │ │ │ ├── twins_pcpvt_large_8xb128_ep300.py │ │ │ ├── twins_pcpvt_small_8xb128_ep300.py │ │ │ ├── twins_svt_base_8xb128_ep300.py │ │ │ ├── twins_svt_large_8xb128_ep300.py │ │ │ └── twins_svt_small_8xb128_ep300.py │ │ ├── uniformer │ │ │ ├── README.md │ │ │ ├── uniformer_base_8xb128_ep300.py │ │ │ ├── uniformer_small_8xb128_fp16_ep300.py │ │ │ ├── uniformer_small_plus_8xb128_ep300.py │ │ │ └── uniformer_tiny_8xb128_ep300.py │ │ ├── unireplknet │ │ │ ├── README.md │ │ │ ├── unireplknet_a_8xb128_accu4_ep300.py │ │ │ ├── unireplknet_f_8xb128_accu4_ep300.py │ │ │ ├── unireplknet_n_8xb128_accu4_ep300.py │ │ │ ├── unireplknet_p_8xb128_accu4_ep300.py │ │ │ ├── unireplknet_s_8xb128_accu4_ep300.py │ │ │ └── unireplknet_t_8xb128_accu4_ep300.py │ │ ├── van │ │ │ ├── README.md │ │ │ ├── van_b4_8xb128_ep300.py │ │ │ ├── van_base_8xb128_ep300.py │ │ │ ├── van_base_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── van_large_8xb128_ep300.py │ │ │ ├── van_small_8xb128_ep300.py │ │ │ ├── van_small_rsb_a3_sz160_8xb256_ep100.py │ │ │ ├── van_tiny_8xb128_ep300.py │ │ │ └── van_tiny_rsb_a3_sz160_8xb256_ep100.py │ │ ├── vanillanet │ │ │ ├── README.md │ │ │ ├── vanillanet_10_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_10_adamw_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_5_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_5_adamw_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_6_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_6_adamw_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_7_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_7_adamw_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_8_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_8_adamw_8xb128_fp16_ep300.py │ │ │ ├── vanillanet_9_8xb128_fp16_ep300.py │ │ │ └── vanillanet_9_adamw_8xb128_fp16_ep300.py │ │ ├── vgg │ │ │ ├── README.md │ │ │ ├── vgg11_4xb64_ep100.py │ │ │ ├── vgg11_bn_4xb64_ep100.py │ │ │ ├── vgg13_4xb64_ep100.py │ │ │ ├── vgg13_bn_4xb64_ep100.py │ │ │ ├── vgg16_4xb64_ep100.py │ │ │ ├── vgg16_bn_4xb64_ep100.py │ │ │ ├── vgg19_4xb64_ep100.py │ │ │ └── vgg19_bn_4xb64_ep100.py │ │ └── vision_transformer │ │ │ ├── README.md │ │ │ ├── vit_base_p16_sz224_8xb64_accu8_ep300.py │ │ │ ├── vit_base_p16_sz384_8xb64_accu8_ep300.py │ │ │ ├── vit_base_p32_sz384_8xb64_accu8_ep300.py │ │ │ ├── vit_large_p16_sz384_8xb64_accu8_ep300.py │ │ │ └── vit_small_p16_sz224_8xb256_accu2_ep300.py │ ├── inaturalist2017 │ │ ├── README.md │ │ ├── auto_train_mixups.py │ │ ├── automix │ │ │ ├── r18_l2_a2_near_lam_cat_mb_mlr1e_3_bb_mlr0_4xb64_ep100.py │ │ │ ├── r50_l2_a2_near_lam_cat_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ │ │ └── rx101_l2_a2_near_lam_cat_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ │ ├── mixups │ │ │ ├── r18_mixups_CE_none_4xb64.py │ │ │ ├── r50_mixups_CE_none_4xb64.py │ │ │ └── rx101_mixups_CE_none_4xb64.py │ │ └── samix │ │ │ ├── r18_l2_a2_bili_val_dp01_mb_mlr1e_3_bb_mlr0_4xb64_ep100.py │ │ │ ├── r50_l2_a2_bili_val_dp01_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ │ │ └── rx101_l2_a2_bili_val_dp01_mb_mlr1e_4_bb_mlr0_4xb64.py │ ├── inaturalist2018 │ │ ├── README.md │ │ ├── auto_train_mixups.py │ │ ├── automix │ │ │ ├── r50_l2_a2_near_lam_cat_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ │ │ └── rx101_l2_a2_near_lam_cat_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ │ ├── mixups │ │ │ ├── r18_mixups_CE_none_4xb64.py │ │ │ ├── r50_mixups_CE_none_4xb64.py │ │ │ └── rx101_mixups_CE_none_4xb64.py │ │ └── samix │ │ │ ├── r50_l2_a2_bili_val_dp01_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ │ │ └── rx101_l2_a2_bili_val_dp01_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ ├── place205 │ │ ├── README.md │ │ ├── auto_train_mixups.py │ │ ├── automix │ │ │ ├── r18_l2_a2_near_lam_cat_mb_mlr1e_3_bb_mlr0_4xb64_ep100.py │ │ │ └── r50_l2_a2_near_lam_cat_mb_mlr1e_3_bb_mlr0_4xb64_ep100.py │ │ ├── mixups │ │ │ ├── r18_mixups_CE_none_4xb64.py │ │ │ └── r50_mixups_CE_none_4xb64.py │ │ └── samix │ │ │ ├── r18_l2_a2_bili_val_dp01_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ │ │ └── r50_l2_a2_bili_val_dp01_mb_mlr1e_4_bb_mlr0_4xb64_ep100.py │ └── tiny_imagenet │ │ ├── README.md │ │ ├── adautomix │ │ └── basic │ │ │ ├── r18_adautomix_blinear.py │ │ │ └── rx50_adautomix_bilinear.py │ │ ├── automix │ │ └── basic │ │ │ ├── r18_l2_a2_near_mb_mlr1e_3_bb_mlr5e_2.py │ │ │ └── rx50_l2_a2_near_mb_mlr1e_3_bb_mlr5e_2_2xb50.py │ │ ├── mixups │ │ ├── auto_train_mixups.py │ │ └── basic │ │ │ ├── r18 │ │ │ ├── mix_modealignmix │ │ │ │ └── r18_mixups_CE_soft_alpha2_ep400.py │ │ │ ├── mix_modeattentivemix │ │ │ │ └── r18_attentivemix_CE_none_alpha2_ep400.py │ │ │ ├── mix_modemixup │ │ │ │ └── r18_mixups_CE_none_alpha1_ep400.py │ │ │ └── mix_modepuzzlemix │ │ │ │ └── r18_mixups_CE_soft_alpha1_ep400.py │ │ │ ├── r18_alignmix_CE_none.py │ │ │ ├── r18_attentivemix_CE_none.py │ │ │ ├── r18_mixups_CE_none.py │ │ │ ├── r18_mixups_CE_none_multi_mode.py │ │ │ ├── r18_mixups_CE_sigm.py │ │ │ ├── r18_mixups_CE_soft.py │ │ │ ├── r18_puzzlemix_CE_soft.py │ │ │ ├── rx50_mixups_CE_none.py │ │ │ └── rx50_mixups_CE_soft.py │ │ └── samix │ │ └── basic │ │ ├── r18 │ │ └── unsampling_modebilinear │ │ │ └── r18_l2_a2_bili_val_dp0_mul_mb_mlr1e_3_bb_mlr5e_2_ep400.py │ │ ├── r18_l2_a2_bili_val_dp0_mul_mb_mlr1e_3_bb_mlr5e_2.py │ │ ├── rx50 │ │ └── unsampling_modebilinear │ │ │ └── rx50_l2_a2_bili_val_dp0_mul_mb_mlr1e_3_bb_mlr5e_2_2xb50_lam_mul_k0_25_mask_adjust0_25_ep400.py │ │ └── rx50_l2_a2_bili_val_dp0_mul_mb_mlr1e_3_bb_mlr5e_2_2xb50.py ├── regression │ ├── _base_ │ │ ├── datasets │ │ │ ├── agedb │ │ │ │ ├── basic_sz224_4xbs64.py │ │ │ │ └── randaug_sz224_4xbs64.py │ │ │ ├── imdb_wiki │ │ │ │ ├── basic_sz224_4xbs64.py │ │ │ │ └── randaug_sz224_4xbs64.py │ │ │ └── rcfmnist │ │ │ │ └── sz32_bs100.py │ │ ├── default_runtime.py │ │ └── models │ │ │ ├── convnext │ │ │ └── convnext_tiny.py │ │ │ ├── resnet │ │ │ ├── resnet18.py │ │ │ ├── resnet18_cifar.py │ │ │ └── resnet50.py │ │ │ └── vision_transformer │ │ │ └── vit_small_p16_sz224.py │ ├── agedb │ │ ├── basic │ │ │ ├── convnext_t_l1_sz224_adamw_4xbs64_ep400.py │ │ │ └── r50_l1_sz224_adamw_4xbs64_ep400.py │ │ └── vits │ │ │ ├── convnext_t_l1_sz224_adamw_4xbs64_ep400.py │ │ │ └── vit_s_l1_sz224_adamw_4xbs64_ep400.py │ ├── imdb_wiki │ │ ├── basic │ │ │ ├── convnext_t_l1_sz224_adamw_4xbs64_ep200.py │ │ │ └── r50_l1_sz224_adamw_4xbs64_ep200.py │ │ └── vits │ │ │ ├── convnext_t_l1_sz224_adamw_4xbs64_ep200.py │ │ │ └── vit_s_l1_sz224_adamw_4xbs64_ep200.py │ └── rcfmnist │ │ └── basic │ │ ├── convnext_t_l1_sz32_adam_bs100_ep400.py │ │ ├── r18_l1_sz32_adam_bs100_ep400.py │ │ └── r50_l1_sz32_adam_bs100_ep400.py ├── selfsup │ ├── _base_ │ │ ├── datasets │ │ │ ├── cifar10 │ │ │ │ ├── a2mim_rgb_m_sz224_bs64.py │ │ │ │ ├── byol_sz224_bs256.py │ │ │ │ ├── deepcluster_sz224_bs64.py │ │ │ │ ├── mae_sz224_bs64.py │ │ │ │ ├── metrics_knn_svm_umap.py │ │ │ │ ├── metrics_umap.py │ │ │ │ ├── mocov1_sz224_bs64.py │ │ │ │ ├── mocov2_sz224_bs64.py │ │ │ │ ├── mocov3_vit_sz224_bs64.py │ │ │ │ ├── simclr_sz224_bs64.py │ │ │ │ ├── simmim_sz192_bs64.py │ │ │ │ ├── simmim_sz224_bs64.py │ │ │ │ └── simsiam_sz224_bs64.py │ │ │ ├── cifar100 │ │ │ │ ├── a2mim_rgb_m_sz224_bs64.py │ │ │ │ ├── byol_sz224_bs256.py │ │ │ │ ├── deepcluster_sz224_bs64.py │ │ │ │ ├── mae_sz224_bs64.py │ │ │ │ ├── metrics_knn_svm_umap.py │ │ │ │ ├── metrics_umap.py │ │ │ │ ├── mocov1_sz224_bs64.py │ │ │ │ ├── mocov2_sz224_bs64.py │ │ │ │ ├── mocov3_vit_sz224_bs64.py │ │ │ │ ├── simclr_sz224_bs64.py │ │ │ │ ├── simmim_sz224_bs64.py │ │ │ │ └── simsiam_sz224_bs64.py │ │ │ ├── imagenet │ │ │ │ ├── a2mim_rgb_m_sz224_bs64.py │ │ │ │ ├── a2mim_rgb_m_sz224_rrc08_bs256.py │ │ │ │ ├── a2mim_rgb_m_sz224_rrc08_bs64.py │ │ │ │ ├── beit_sz224_bs64.py │ │ │ │ ├── byol_sz224_bs64.py │ │ │ │ ├── cae_sz224_bs64.py │ │ │ │ ├── deepcluster_sz224_bs64.py │ │ │ │ ├── dino_mcrop-2-8_sz224_96_bs64.py │ │ │ │ ├── mae_sz224_bs256.py │ │ │ │ ├── metrics_knn_umap.py │ │ │ │ ├── metrics_umap.py │ │ │ │ ├── mim_feat_sz224_p16_bs64.py │ │ │ │ ├── mim_feat_sz224_p32_bs64.py │ │ │ │ ├── mocov1_sz224_bs64.py │ │ │ │ ├── mocov2_sz224_bs64.py │ │ │ │ ├── mocov3_cnn_sz224_bs64.py │ │ │ │ ├── mocov3_vit_sz224_bs64.py │ │ │ │ ├── npid_sz224_bs64.py │ │ │ │ ├── odc_sz224_bs64.py │ │ │ │ ├── relative-loc_sz224_bs64.py │ │ │ │ ├── rotation-pred_sz224_bs64.py │ │ │ │ ├── simclr_sz224_bs64.py │ │ │ │ ├── simmim_sz192_bs64.py │ │ │ │ ├── simmim_sz224_bs64.py │ │ │ │ ├── simmim_sz224_p16_bs64.py │ │ │ │ └── swav_mcrop-2-6_sz224_96_bs32.py │ │ │ ├── imagenet100 │ │ │ │ ├── a2mim_rgb_m_sz192_bs64.py │ │ │ │ ├── a2mim_rgb_m_sz224_bs64.py │ │ │ │ ├── byol_sz224_bs64.py │ │ │ │ ├── deepcluster_sz224_bs64.py │ │ │ │ ├── mae_sz224_bs64.py │ │ │ │ ├── metrics_knn_umap.py │ │ │ │ ├── metrics_umap.py │ │ │ │ ├── mim_feat_sz224_p16_bs64.py │ │ │ │ ├── mocov1_sz224_bs64.py │ │ │ │ ├── mocov2_sz224_bs64.py │ │ │ │ ├── mocov3_cnn_sz224_bs64.py │ │ │ │ ├── mocov3_vit_sz224_bs64.py │ │ │ │ ├── npid_sz224_bs64.py │ │ │ │ ├── odc_sz224_bs64.py │ │ │ │ ├── relative-loc_sz224_bs64.py │ │ │ │ ├── rotation-pred_sz224_bs64.py │ │ │ │ ├── simclr_sz224_bs64.py │ │ │ │ ├── simmim_sz192_bs64.py │ │ │ │ └── swav_mcrop-2-6_sz224_96_bs32.py │ │ │ └── stl10 │ │ │ │ ├── a2mim_rgb_m_sz224_bs64.py │ │ │ │ ├── byol_sz96_bs256.py │ │ │ │ ├── mocov1_sz96_bs64.py │ │ │ │ ├── mocov2_sz96_bs64.py │ │ │ │ ├── simclr_sz96_bs256.py │ │ │ │ └── simmim_sz224_bs64.py │ │ ├── default_runtime.py │ │ └── models │ │ │ ├── a2mim │ │ │ ├── convnext_b.py │ │ │ ├── convnext_s.py │ │ │ ├── convnext_t.py │ │ │ ├── deit_small.py │ │ │ ├── moganet_s.py │ │ │ ├── r101.py │ │ │ ├── r152.py │ │ │ ├── r200.py │ │ │ ├── r50.py │ │ │ ├── swin_base.py │ │ │ ├── swin_tiny.py │ │ │ ├── vit_base.py │ │ │ ├── vit_large.py │ │ │ └── vit_small.py │ │ │ ├── barlowtwins │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── beit │ │ │ ├── deit_small.py │ │ │ └── vit_base.py │ │ │ ├── byol │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── cae │ │ │ └── vit_base.py │ │ │ ├── deepcluster │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── densecl │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── dino │ │ │ └── vit_base.py │ │ │ ├── mae │ │ │ ├── vit_base.py │ │ │ ├── vit_huge.py │ │ │ ├── vit_large.py │ │ │ └── vit_small.py │ │ │ ├── maskfeat │ │ │ ├── deit_small.py │ │ │ ├── r50.py │ │ │ ├── swin_base.py │ │ │ └── vit_base.py │ │ │ ├── mocov1 │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── mocov2 │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── mocov3 │ │ │ ├── r50.py │ │ │ └── vit_small.py │ │ │ ├── npid │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── odc │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── relative_loc │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── rotation_pred │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── simclr │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ ├── simmim │ │ │ ├── convnext_b.py │ │ │ ├── convnext_s.py │ │ │ ├── convnext_t.py │ │ │ ├── deit_small.py │ │ │ ├── r101.py │ │ │ ├── r152.py │ │ │ ├── r200.py │ │ │ ├── r50.py │ │ │ ├── swin_base.py │ │ │ ├── swin_tiny.py │ │ │ ├── vit_base.py │ │ │ ├── vit_large.py │ │ │ └── vit_small.py │ │ │ ├── simsiam │ │ │ ├── r18.py │ │ │ └── r50.py │ │ │ └── swav │ │ │ ├── r18.py │ │ │ └── r50.py │ ├── a2mim │ │ ├── README.md │ │ ├── cifar100 │ │ │ ├── r50_l3_sz224_init_4xb256_accu2_cos_ep1000.py │ │ │ └── vit_small_l0_sz224_4xb256_accu2_cos_ep1000.py │ │ ├── imagenet │ │ │ ├── convnext_b_l3_sz224_init_8xb256_cos_ep300.py │ │ │ ├── convnext_s_l3_sz224_init_8xb256_cos_ep300.py │ │ │ ├── convnext_t_l3_sz224_init_8xb256_cos_ep300.py │ │ │ ├── moganet_s_l3_sz224_init_8xb256_cos_ep300.py │ │ │ ├── r101_l3_sz224_init_8xb256_cos_ep300.py │ │ │ ├── r152_l3_sz224_init_8xb256_cos_ep300.py │ │ │ ├── r200_l3_sz224_init_8xb256_cos_ep300.py │ │ │ ├── r50_l0_sz224_8xb256_cos_ep100.py │ │ │ ├── r50_l3_sz224_8xb256_cos_ep100.py │ │ │ ├── r50_l3_sz224_init_8xb256_cos_ep100.py │ │ │ ├── r50_l3_sz224_init_8xb256_cos_ep300.py │ │ │ ├── vit_base_l0_sz224_8xb128_accu2_step_ep800.py │ │ │ ├── vit_large_l0_sz224_8xb128_accu2_step_ep800.py │ │ │ └── vit_small_l0_sz224_8xb256_step_ep800.py │ │ └── imagenet100 │ │ │ ├── deit_small_l0_mlp_sz192_4xb256_accu2_cos_ep800.py │ │ │ ├── deit_small_l8_sz192_4xb256_accu2_cos_ep800.py │ │ │ └── r50_l3_sz192_4xb256_accu2_cos_fp16_ep800.py │ ├── barlowtwins │ │ ├── README.md │ │ ├── cifar10 │ │ │ ├── r18_8xb256_cos_lr1_6_fp16_ep1000.py │ │ │ └── r18_8xb256_cos_lr1_6_knn_svm_umap_fp16_ep1000.py │ │ ├── cifar100 │ │ │ ├── r18_8xb256_cos_lr1_6_fp16_ep1000.py │ │ │ └── r18_8xb256_cos_lr1_6_knn_svm_umap_fp16_ep1000.py │ │ ├── imagenet │ │ │ ├── r50_8xb64_accu4_cos_lr1_6.py │ │ │ ├── r50_8xb64_accu4_cos_lr1_6_ep1000.py │ │ │ └── r50_8xb64_accu4_cos_lr1_6_ep300.py │ │ └── imagenet100 │ │ │ └── r18_8xb64_accu4_cos_lr1_6_fp16_ep1000.py │ ├── beit │ │ ├── RAEDME.md │ │ └── imagenet │ │ │ └── vit_base_sz224_8xb64_accu4_cos_fp16_ep800.py │ ├── byol │ │ ├── README.md │ │ ├── cifar10 │ │ │ ├── r18_8xb256_accu2_cos_lr4_8_fp16_ep1000.py │ │ │ └── r18_8xb256_accu2_cos_lr4_8_knn_svm_umap_fp16_ep1000.py │ │ ├── cifar100 │ │ │ ├── r18_8xb256_accu2_cos_lr4_8_fp16_ep1000.py │ │ │ └── r18_8xb256_accu2_cos_lr4_8_knn_svm_umap_fp16_ep1000.py │ │ ├── imagenet │ │ │ ├── r18_8xb64_accu8_cos_lr7_2_ep100.py │ │ │ ├── r18_8xb64_accu8_cos_lr7_2_knn_umap_ep100.py │ │ │ ├── r50_8xb64_accu8_cos_lr4_8.py │ │ │ ├── r50_8xb64_accu8_cos_lr4_8_fp16_ep200.py │ │ │ ├── r50_8xb64_accu8_cos_lr4_8_fp16_ep300.py │ │ │ └── r50_8xb64_accu8_cos_lr7_2_ep100.py │ │ ├── imagenet100 │ │ │ ├── r18_8xb64_accu8_cos_lr4_8_fp16_ep400.py │ │ │ └── r18_8xb64_accu8_cos_lr4_8_knn_umap_fp16_ep400.py │ │ └── stl10 │ │ │ └── r50_8xb256_accu2_cos_lr4_8_fp16_ep1000.py │ ├── cae │ │ ├── RAEDME.md │ │ └── imagenet │ │ │ ├── vit_base_sz224_8xb256_cos_fp16_ep300.py │ │ │ └── vit_base_sz224_8xb64_accu4_cos_fp16_ep300.py │ ├── deepcluster │ │ ├── README.md │ │ ├── cifar10 │ │ │ ├── r18_8xb64_step_ep1000.py │ │ │ └── r18_8xb64_step_knn_svm_umap_ep1000.py │ │ ├── cifar100 │ │ │ ├── r18_8xb64_step_ep1000.py │ │ │ └── r18_8xb64_step_knn_svm_umap_ep1000.py │ │ └── imagenet │ │ │ ├── r18_8xb64_step_ep100.py │ │ │ ├── r50_8xb64_step_ep200.py │ │ │ └── r50_sobel_8xb64_step_ep200.py │ ├── densecl │ │ ├── README.md │ │ ├── cifar10 │ │ │ └── r18_4xb64_cos_ep1000.py │ │ ├── cifar100 │ │ │ └── r18_4xb64_cos_ep1000.py │ │ └── imagenet │ │ │ ├── r18_4xb64_cos_ep100.py │ │ │ ├── r50_4xb64_cos.py │ │ │ └── r50_4xb64_cos_fp16_ep200.py │ ├── dino │ │ ├── README.md │ │ └── imagenet │ │ │ └── vit_base_8xb64_accu8_cos_fp16_ep100.py │ ├── mae │ │ ├── README.md │ │ ├── cifar10 │ │ │ ├── vit_small_dec6_dim512_8xb128_accu4_cos_ep1000.py │ │ │ └── vit_small_dec6_dim512_8xb128_accu4_cos_umap_fp16_ep1000.py │ │ ├── cifar100 │ │ │ ├── vit_small_dec6_dim512_8xb128_accu4_cos_ep1000.py │ │ │ └── vit_small_dec6_dim512_8xb128_accu4_cos_umap_fp16_ep1000.py │ │ ├── imagenet │ │ │ ├── vit_base_dec8_dim512_8xb512_cos.py │ │ │ ├── vit_base_dec8_dim512_8xb512_cos_fp16_ep1600.py │ │ │ ├── vit_base_dec8_dim512_8xb512_cos_fp16_ep400.py │ │ │ ├── vit_base_dec8_dim512_8xb512_cos_fp16_ep800.py │ │ │ ├── vit_huge_dec8_dim512_8xb128_accu4_cos_fp16_ep1600.py │ │ │ ├── vit_large_dec8_dim512_8xb256_accu2_cos.py │ │ │ ├── vit_large_dec8_dim512_8xb256_accu2_cos_fp16_ep1600.py │ │ │ ├── vit_large_dec8_dim512_8xb256_accu2_cos_fp16_ep400.py │ │ │ ├── vit_large_dec8_dim512_8xb256_accu2_cos_fp16_ep800.py │ │ │ └── vit_small_dec6_dim512_8xb512_cos.py │ │ └── imagenet100 │ │ │ ├── vit_small_dec6_dim512_8xb128_accu4_cos_ep400.py │ │ │ └── vit_small_dec6_dim512_8xb128_accu4_cos_umap_ep400.py │ ├── maskfeat │ │ ├── README.md │ │ ├── imagenet │ │ │ ├── swin_tiny_hog_108_sz224_8xb128_accu2_cos_ep100.py │ │ │ ├── swin_tiny_hog_9_sz224_8xb128_accu2_cos_ep100.py │ │ │ ├── vit_base_hog_108_sz224_8xb128_accu2_cos_fp16_ep300.py │ │ │ └── vit_base_hog_108_sz224_8xb128_accu2_cos_fp16_ep800.py │ │ └── imagenet100 │ │ │ ├── deit_small_hog_108_sz224_4xb256_accu2_cos_ep100.py │ │ │ ├── deit_small_hog_9_sz224_4xb256_accu2_cos_ep800.py │ │ │ └── deit_small_sobel_sz224_4xb256_accu2_cos_ep100.py │ ├── mocov1 │ │ ├── README.md │ │ ├── cifar10 │ │ │ └── r18_4xb64_step_ep1000.py │ │ ├── cifar100 │ │ │ └── r18_4xb64_step_ep1000.py │ │ └── imagenet │ │ │ ├── r18_4xb64_step_ep100.py │ │ │ └── r50_4xb64_step_ep200.py │ ├── mocov2 │ │ ├── README.md │ │ ├── cifar10 │ │ │ ├── r18_4xb64_cos_ep1000.py │ │ │ └── r18_4xb64_cos_umap_knn_ep1000.py │ │ ├── cifar100 │ │ │ ├── r18_4xb64_cos_ep1000.py │ │ │ └── r18_4xb64_cos_umap_knn_ep1000.py │ │ ├── imagenet │ │ │ ├── r18_4xb64_cos_ep100.py │ │ │ ├── r50_4xb64_cos.py │ │ │ └── r50_4xb64_cos_fp16_ep200.py │ │ └── stl10 │ │ │ ├── r50_4xb64_cos_ep1000.py │ │ │ └── r50_4xb64_cos_umap_knn_ep1000.py │ ├── mocov3 │ │ ├── README.md │ │ ├── cifar10 │ │ │ ├── vit_small_8xb64_accu8_cos_ep1000.py │ │ │ └── vit_small_8xb64_accu8_cos_umap_knn_fp16_ep1000.py │ │ ├── cifar100 │ │ │ ├── vit_small_8xb64_accu8_cos_ep1000.py │ │ │ └── vit_small_8xb64_accu8_cos_umap_knn_fp16_ep1000.py │ │ ├── imagenet │ │ │ ├── r50_4xb256_accu4_cos_fp16.py │ │ │ ├── vit_small_8xb64_accu8_cos.py │ │ │ └── vit_small_8xb64_accu8_cos_fp16_ep300.py │ │ ├── imagenet100 │ │ │ ├── r50_4xb256_accu4_cos_fp16_ep400.py │ │ │ ├── r50_4xb256_accu4_cos_knn_umap_fp16_ep400.py │ │ │ ├── vit_small_8xb64_accu8_cos_ep400.py │ │ │ └── vit_small_8xb64_accu8_cos_knn_umap_fp16_ep400.py │ │ └── stl10 │ │ │ ├── vit_small_8xb256_accu2_cos_ep1000.py │ │ │ └── vit_small_8xb256_accu2_cos_knn_umap_ep1000.py │ ├── npid │ │ ├── README.md │ │ └── imagenet │ │ │ ├── r18_4xb64_step_ep100.py │ │ │ ├── r50_4xb64_step_ep200.py │ │ │ └── r50_ensure_neg_4xb64_step_ep200.py │ ├── odc │ │ ├── README.md │ │ └── imagenet │ │ │ ├── r18_8xb64_step_ep100.py │ │ │ └── r50_8xb64_step_ep440.py │ ├── relative_loc │ │ ├── README.md │ │ └── imagenet │ │ │ ├── r18_8xb64_step_ep100.py │ │ │ └── r50_8xb64_step_ep70.py │ ├── rotation_pred │ │ ├── README.md │ │ └── imagenet │ │ │ ├── r18_8xb64_step_ep100.py │ │ │ └── r50_8xb64_step_ep70.py │ ├── simclr │ │ ├── README.md │ │ ├── cifar10 │ │ │ ├── r18_8xb256_cos_lr2_4_fp16_ep1000.py │ │ │ └── r18_8xb256_cos_lr2_4_knn_svm_umap_fp16_ep1000.py │ │ ├── cifar100 │ │ │ ├── r18_8xb256_cos_lr2_4_fp16_ep1000.py │ │ │ └── r18_8xb256_cos_lr2_4_knn_svm_umap_fp16_ep1000.py │ │ └── imagenet │ │ │ ├── r18_8xb64_cos_lr0_6_ep100.py │ │ │ ├── r50_16xb256_cos_lr4_8_fp16_ep200.py │ │ │ ├── r50_8xb64_cos_lr0_6.py │ │ │ └── r50_8xb64_cos_lr0_6_fp16_ep200.py │ ├── simmim │ │ ├── README.md │ │ ├── cifar10 │ │ │ ├── swin_base_sz192_8xb64_accu4_cos_ep1000.py │ │ │ └── swin_base_sz192_8xb64_accu4_cos_umap_ep1000.py │ │ ├── cifar100 │ │ │ ├── swin_base_sz192_8xb64_accu4_cos_ep1000.py │ │ │ └── swin_base_sz192_8xb64_accu4_cos_umap_ep1000.py │ │ ├── imagenet │ │ │ ├── convnext_base_sz224_8xb256_fp16_ep300.py │ │ │ ├── convnext_small_sz224_8xb256_fp16_ep300.py │ │ │ ├── convnext_tiny_sz224_8xb256_fp16_ep300.py │ │ │ ├── r101_sz224_8xb256_fp16_ep300.py │ │ │ ├── r152_sz224_8xb256_fp16_ep300.py │ │ │ ├── r200_sz224_8xb256_fp16_ep300.py │ │ │ ├── r50_sz224_8xb256_fp16_ep300.py │ │ │ ├── swin_base_sz192_8xb128_accu2_cos_ep100.py │ │ │ ├── swin_base_sz192_8xb128_accu2_cos_ep800.py │ │ │ ├── vit_base_rgb_m_sz224_8xb128_accu2_step_fp16_ep800.py │ │ │ ├── vit_base_sz224_8xb128_accu2_step_fp16_ep800.py │ │ │ ├── vit_large_sz224_8xb128_accu2_step_fp16_ep800.py │ │ │ └── vit_small_sz224_8xb256_step_fp16_ep800.py │ │ └── imagenet100 │ │ │ ├── r50_sz192_8xb128_accu2_cos_ep800.py │ │ │ └── swin_tiny_sz192_8xb128_accu2_cos_ep800.py │ ├── simsiam │ │ ├── README.md │ │ ├── cifar100 │ │ │ ├── r18_8xb64_cos_lr0_1_fp16_ep1000.py │ │ │ └── r18_8xb64_cos_lr0_1_knn_svm_umap_fp16_ep1000.py │ │ └── imagenet │ │ │ ├── r18_8xb64_cos_lr0_1_ep100.py │ │ │ ├── r18_8xb64_cos_lr0_1_knn_umap_ep100.py │ │ │ ├── r50_4xb64_cos_lr0_05.py │ │ │ ├── r50_4xb64_cos_lr0_05_ep100.py │ │ │ ├── r50_4xb64_cos_lr0_05_ep200.py │ │ │ ├── r50_8xb64_cos_lr0_1.py │ │ │ └── r50_8xb64_cos_lr0_1_ep100.py │ └── swav │ │ ├── README.md │ │ └── imagenet │ │ ├── r18_4xb64_cos_lr0_6-mcrop-224_2-96_6_ep100.py │ │ ├── r18_8xb64_accu8_cos_lr9_6-mcrop-224_2-96_6_ep100.py │ │ ├── r50_4xb64_cos_lr0_6-mcrop-224_2-96_6.py │ │ ├── r50_8xb64_accu8_cos_lr9_6-mcrop-224_2-96_6.py │ │ └── r50_8xb64_accu8_cos_lr9_6-mcrop-224_2-96_6_fp16_ep200.py └── semisup │ ├── _base_ │ ├── datasets │ │ ├── aircrafts │ │ │ ├── per_15_sz224_bs24.py │ │ │ ├── per_30_sz224_bs24.py │ │ │ └── per_50_sz224_bs24.py │ │ ├── cars │ │ │ ├── per_15_sz224_bs24.py │ │ │ ├── per_30_sz224_bs24.py │ │ │ └── per_50_sz224_bs24.py │ │ ├── cifar100 │ │ │ ├── 10k_randaug_sz32_bs64.py │ │ │ ├── 2_5k_randaug_sz32_bs64.py │ │ │ └── 400_randaug_sz32_bs64.py │ │ └── cub200 │ │ │ ├── per_15_sz224_bs24.py │ │ │ ├── per_30_sz224_bs24.py │ │ │ └── per_50_sz224_bs24.py │ └── default_runtime.py │ ├── dmixmatch │ └── cifar100 │ │ └── wrn28_8 │ │ └── wrn28_8_10k_bs64_7_cos_randaug_ema_p1.py │ ├── fixmatch │ ├── cifar100 │ │ └── wrn28_8 │ │ │ ├── wrn28_8_10k_bs64_7_cos_randaug_ema_p1_step_m0_99_cos_dedup_minlr1_5_ep2400.py │ │ │ ├── wrn28_8_2_5k_bs64_7_cos_randaug_ema_p1_step_m0_99_cos_dedup_minlr1_5_ep2400.py │ │ │ └── wrn28_8_400_bs64_7_cos_randaug_ema_p1_step_m0_99_cos_dedup_minlr1_5_ep2400.py │ └── imagenet │ │ └── r50 │ │ ├── r50_100per_cls_bs64x4_7_cos_randaug_ema_p1_step_m0_999_p072_cos_minlr1_5_ep400.py │ │ ├── r50_10per_bs64x4_7_cos_randaug_ema_p1_step_m0_999_p072_cos_minlr1_5_ep400.py │ │ └── r50_1per_bs64x4_7_cos_randaug_ema_p1_step_m0_999_p072_cos_minlr1_5_ep400.py │ └── selftuning │ ├── aircrafts │ ├── r18 │ │ ├── proj_dim_1024 │ │ │ └── r18_per_15_bs24_cos_no_randaug_weight_decay0_0001_nesterovTrue_ep120.py │ │ ├── r18_per_15_bs24_cos_no_randaug.py │ │ ├── r18_per_30_bs24_cos_no_randaug.py │ │ └── r18_per_50_bs24_cos_no_randaug.py │ └── r50 │ │ ├── r50_per_15_bs24_cos_no_randaug.py │ │ ├── r50_per_30_bs24_cos_no_randaug.py │ │ └── r50_per_50_bs24_cos_no_randaug.py │ ├── auto_train_self_tuning.py │ ├── cars │ ├── r18 │ │ ├── proj_dim_1024 │ │ │ └── r18_per_15_bs24_cos_no_randaug_weight_decay0_0001_nesterovTrue_ep120.py │ │ ├── r18_per_15_bs24_cos_no_randaug.py │ │ ├── r18_per_30_bs24_cos_no_randaug.py │ │ └── r18_per_50_bs24_cos_no_randaug.py │ └── r50 │ │ ├── r50_per_15_bs24_cos_no_randaug.py │ │ ├── r50_per_30_bs24_cos_no_randaug.py │ │ └── r50_per_50_bs24_cos_no_randaug.py │ └── cub200 │ ├── r18 │ ├── proj_dim_1024 │ │ └── r18_per_15_bs24_cos_no_randaug_weight_decay0_0001_nesterovTrue_ep120.py │ ├── r18_per_15_bs24_cos_no_randaug.py │ ├── r18_per_30_bs24_cos_no_randaug.py │ └── r18_per_50_bs24_cos_no_randaug.py │ └── r50 │ ├── r50_per_15_bs24_cos_no_randaug.py │ ├── r50_per_30_bs24_cos_no_randaug.py │ └── r50_per_50_bs24_cos_no_randaug.py ├── demo ├── bird.JPEG └── cat-dog.png ├── docker └── Dockerfile ├── docs └── en │ ├── Makefile │ ├── _static │ ├── css │ │ └── readthedocs.css │ └── image │ │ └── logo.png │ ├── _templates │ ├── 404.html │ ├── autosummary │ │ └── class.rst │ ├── callable.rst │ └── data_transform.rst │ ├── api.rst │ ├── api │ ├── apis.rst │ ├── core.rst │ ├── datasets.rst │ ├── models.rst │ ├── models.utils.rst │ ├── third_party.rst │ └── utils.rst │ ├── awesome │ ├── awesome_mixup.md │ └── awesome_selfsup.md │ ├── changelog.md │ ├── compatibility.md │ ├── conf.py │ ├── faq.md │ ├── get_started.md │ ├── index.rst │ ├── install.md │ ├── mixup_benchmarks │ ├── Mixup_cifar.md │ ├── Mixup_downstream.md │ └── Mixup_imagenet.md │ ├── model_zoos │ ├── Mixup_sup.md │ ├── Model_Zoo_selfsup.md │ ├── Model_Zoo_semisup.md │ └── Model_Zoo_sup.md │ ├── switch_language.md │ ├── tools │ ├── analysis.md │ ├── pytorch2onnx.md │ ├── pytorch2torchscript.md │ └── visualization.md │ └── tutorials │ ├── 0_config.md │ ├── 1_new_dataset.md │ ├── 2_data_pipeline.md │ ├── 3_new_modules.md │ ├── 4_schedule.md │ ├── 5_runtime.md │ ├── index.rst │ ├── ssl_classification.md │ ├── ssl_detection.md │ └── ssl_segmentation.md ├── openmixup ├── __init__.py ├── apis │ ├── __init__.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── export │ │ ├── __init__.py │ │ └── test.py │ ├── hooks │ │ ├── __init__.py │ │ ├── addtional_scheduler.py │ │ ├── builder.py │ │ ├── deepcluster_automix_hook.py │ │ ├── deepcluster_hook.py │ │ ├── ema_hook.py │ │ ├── extractor.py │ │ ├── lr_scheduler.py │ │ ├── momentum_hook.py │ │ ├── odc_hook.py │ │ ├── optimizer_hook.py │ │ ├── precise_bn_hook.py │ │ ├── registry.py │ │ ├── save_hook.py │ │ ├── selfsup_metric_hook.py │ │ ├── swav_hook.py │ │ └── validate_hook.py │ └── optimizers │ │ ├── __init__.py │ │ ├── adabelief.py │ │ ├── adabound.py │ │ ├── adafactor.py │ │ ├── adahessian.py │ │ ├── adamp.py │ │ ├── adan.py │ │ ├── builder.py │ │ ├── constructor.py │ │ ├── lamb.py │ │ ├── lars.py │ │ ├── lion.py │ │ ├── madgrad.py │ │ ├── nvnovograd.py │ │ ├── sam.py │ │ ├── sgdp.py │ │ └── sophia.py ├── datasets │ ├── __init__.py │ ├── base.py │ ├── beit_mim.py │ ├── builder.py │ ├── classification.py │ ├── contrastive.py │ ├── data_sources │ │ ├── __init__.py │ │ ├── cifar.py │ │ ├── image_list.py │ │ ├── imagenet.py │ │ └── mnist.py │ ├── dataset_wrappers.py │ ├── deepcluster.py │ ├── deepcluster_contrastive.py │ ├── extraction.py │ ├── loader │ │ ├── __init__.py │ │ ├── build_loader.py │ │ └── sampler.py │ ├── masked_image.py │ ├── multi_view.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── auto_augment.py │ │ ├── compose.py │ │ └── transforms.py │ ├── registry.py │ ├── regression.py │ ├── relative_loc.py │ ├── rotation_pred.py │ ├── semi_supervised.py │ └── utils.py ├── models │ ├── __init__.py │ ├── augments │ │ ├── __init__.py │ │ ├── alignmix.py │ │ ├── attentivemix.py │ │ ├── augmix.py │ │ ├── cutmix.py │ │ ├── fmix.py │ │ ├── gridmix.py │ │ ├── guidedmix.py │ │ ├── mixpro.py │ │ ├── mixup.py │ │ ├── puzzlemix.py │ │ ├── resizemix.py │ │ ├── saliencymix.py │ │ ├── smmix.py │ │ ├── smoothmix.py │ │ ├── snapmix.py │ │ ├── tla.py │ │ ├── tokenmix.py │ │ └── transmix.py │ ├── backbones │ │ ├── __init__.py │ │ ├── alexnet.py │ │ ├── base_backbone.py │ │ ├── beit.py │ │ ├── context_cluster.py │ │ ├── convmixer.py │ │ ├── convnext.py │ │ ├── cspnet.py │ │ ├── davit.py │ │ ├── deit.py │ │ ├── deit3.py │ │ ├── densenet.py │ │ ├── edgenext.py │ │ ├── efficientformer.py │ │ ├── efficientnet.py │ │ ├── efficientnet_v2.py │ │ ├── hornet.py │ │ ├── hrnet.py │ │ ├── inception_v3.py │ │ ├── lenet.py │ │ ├── levit.py │ │ ├── lit.py │ │ ├── metaformer.py │ │ ├── mim_resnet.py │ │ ├── mim_swin.py │ │ ├── mim_vit.py │ │ ├── mlp_mixer.py │ │ ├── mobilenet_v2.py │ │ ├── mobilenet_v3.py │ │ ├── mobileone.py │ │ ├── mobilevit.py │ │ ├── moganet.py │ │ ├── mvit.py │ │ ├── poolformer.py │ │ ├── pvt.py │ │ ├── regnet.py │ │ ├── replknet.py │ │ ├── repmlp.py │ │ ├── repvgg.py │ │ ├── res2net.py │ │ ├── resnest.py │ │ ├── resnet_mmcls.py │ │ ├── resnext.py │ │ ├── revvit.py │ │ ├── riformer.py │ │ ├── rwkv.py │ │ ├── seresnet.py │ │ ├── shufflenet_v1.py │ │ ├── shufflenet_v2.py │ │ ├── starnet.py │ │ ├── swin_transformer.py │ │ ├── swin_transformer_v2.py │ │ ├── t2t_vit.py │ │ ├── timm_backbone.py │ │ ├── transnext.py │ │ ├── twins.py │ │ ├── uniformer.py │ │ ├── unireplknet.py │ │ ├── van.py │ │ ├── vanillanet.py │ │ ├── vgg.py │ │ ├── vig.py │ │ ├── vision_transformer.py │ │ ├── wide_resnet.py │ │ └── xcit.py │ ├── builder.py │ ├── classifiers │ │ ├── __init__.py │ │ ├── adautomix.py │ │ ├── automix.py │ │ ├── base_model.py │ │ ├── classification.py │ │ ├── mergemix.py │ │ └── mixup_classification.py │ ├── heads │ │ ├── __init__.py │ │ ├── adaptive_mask.py │ │ ├── cls_head.py │ │ ├── cls_mixup_head.py │ │ ├── cls_mlp_head.py │ │ ├── contrastive_head.py │ │ ├── dino_head.py │ │ ├── latent_pred_head.py │ │ ├── mim_head.py │ │ ├── multi_cls_head.py │ │ ├── norm_linear_head.py │ │ ├── pmix_block.py │ │ ├── reg_head.py │ │ ├── swav_head.py │ │ ├── tokenizer_head.py │ │ └── vision_transformer_head.py │ ├── losses │ │ ├── __init__.py │ │ ├── arcface_loss.py │ │ ├── asymmetric_loss.py │ │ ├── cross_entropy_loss.py │ │ ├── distillation_loss.py │ │ ├── focal_loss.py │ │ ├── label_smooth_loss.py │ │ ├── regression_loss.py │ │ ├── seesaw_loss.py │ │ ├── semantic_softmax_loss.py │ │ └── utils.py │ ├── memories │ │ ├── __init__.py │ │ ├── odc_memory.py │ │ └── simple_memory.py │ ├── necks │ │ ├── __init__.py │ │ ├── conv_necks.py │ │ ├── hr_necks.py │ │ ├── mim_neck.py │ │ ├── mlp_necks.py │ │ └── transformer_necks.py │ ├── registry.py │ ├── selfsup │ │ ├── __init__.py │ │ ├── a2mim.py │ │ ├── barlowtwins.py │ │ ├── beit.py │ │ ├── byol.py │ │ ├── cae.py │ │ ├── deepcluster.py │ │ ├── densecl.py │ │ ├── dino.py │ │ ├── mae.py │ │ ├── maskfeat.py │ │ ├── moco.py │ │ ├── moco_mix.py │ │ ├── moco_samix.py │ │ ├── mocov3.py │ │ ├── npid.py │ │ ├── odc.py │ │ ├── relative_loc.py │ │ ├── rotation_pred.py │ │ ├── simclr.py │ │ ├── simclr_mix.py │ │ ├── simmim.py │ │ ├── simsiam.py │ │ └── swav.py │ ├── semisup │ │ ├── __init__.py │ │ ├── dmix_tuning.py │ │ ├── dmixmatch.py │ │ ├── fine_tuning.py │ │ ├── fixmatch.py │ │ └── self_tuning.py │ └── utils │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── augmentation.py │ │ ├── evaluation │ │ ├── __init__.py │ │ ├── eval_metrics.py │ │ ├── mean_ap.py │ │ └── multilabel_eval_metrics.py │ │ ├── gather_layer.py │ │ ├── grad_weight.py │ │ ├── helpers.py │ │ ├── layers │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── channel_shuffle.py │ │ ├── conv_ws.py │ │ ├── dall_e.py │ │ ├── drop.py │ │ ├── inverted_residual.py │ │ ├── layer_scale.py │ │ ├── make_divisible.py │ │ ├── norm.py │ │ ├── patch_embed.py │ │ ├── pooling.py │ │ ├── pos_embed.py │ │ ├── res_layer_extra_norm.py │ │ ├── scale.py │ │ ├── se_layer.py │ │ ├── smoothing.py │ │ ├── sobel.py │ │ ├── sparse_modules.py │ │ ├── transformer.py │ │ └── weight_init.py │ │ ├── merge.py │ │ ├── target_generators │ │ ├── __init__.py │ │ ├── clip.py │ │ ├── clip_generator.py │ │ ├── dall_e.py │ │ ├── hog_generator.py │ │ ├── vector_quantizer.py │ │ └── vqkd.py │ │ └── visualization │ │ ├── __init__.py │ │ ├── draw_hog.py │ │ ├── image.py │ │ └── plot_torch.py ├── third_party │ ├── __init__.py │ ├── clustering.py │ ├── distributed_sinkhorn.py │ ├── knn_classifier.py │ └── svm_classifier.py ├── utils │ ├── __init__.py │ ├── alias_multinomial.py │ ├── collect.py │ ├── collect_env.py │ ├── config_tools.py │ ├── contextmanagers.py │ ├── dist_utils.py │ ├── flops_counter.py │ ├── fp16_utils.py │ ├── gather.py │ ├── logger.py │ ├── loss_landscape_utils.py │ ├── misc.py │ ├── profiling.py │ ├── registry.py │ └── setup_env.py └── version.py ├── requirements.txt ├── requirements ├── docs.txt ├── mminstall.txt ├── optional.txt ├── readthedocs.txt ├── runtime.txt └── tests.txt ├── setup.cfg ├── setup.py ├── tests ├── test_datasets │ ├── test_dataset_utils.py │ ├── test_dataset_vis.py │ ├── test_pipelines │ │ └── test_auto_augment.py │ └── test_utils.py ├── test_metrics │ ├── test_eval_metrics.py │ ├── test_loss.py │ └── test_tools.py ├── test_models │ ├── test_backbones │ │ ├── test_convmixer.py │ │ ├── test_convnext.py │ │ ├── test_deit.py │ │ ├── test_densenet.py │ │ ├── test_edgenext.py │ │ ├── test_efficientformer.py │ │ ├── test_efficientnet.py │ │ ├── test_hornet.py │ │ ├── test_mlp_mixer.py │ │ ├── test_mobilenet_v2.py │ │ ├── test_mobilenet_v3.py │ │ ├── test_mvit.py │ │ ├── test_poolformer.py │ │ ├── test_resnest.py │ │ ├── test_resnet_mmcls.py │ │ ├── test_resnext.py │ │ ├── test_seresnet.py │ │ ├── test_shufflenet_v1.py │ │ ├── test_shufflenet_v2.py │ │ ├── test_timm_backbone.py │ │ ├── test_van.py │ │ ├── test_vgg.py │ │ ├── test_vision_transformer.py │ │ └── utils.py │ ├── test_classifiers.py │ ├── test_heads.py │ ├── test_necks.py │ └── test_utils │ │ ├── test_augments.py │ │ └── test_layers │ │ ├── test_attention.py │ │ ├── test_channel_shuffle_division.py │ │ ├── test_inverted_residual.py │ │ ├── test_patch_embed.py │ │ ├── test_pos_embed.py │ │ └── test_se_layer.py ├── test_runtime │ ├── test_eval_hook.py │ ├── test_hooks.py │ ├── test_optimizer.py │ └── test_preciseBN_hook.py └── test_utils │ ├── test_setup_env.py │ └── test_visualization.py └── tools ├── analysis_tools ├── analyze_logs.py ├── calibration_fgsm.py ├── count_parameters.py ├── courrption.py ├── get_flops.py ├── merge_logs.py ├── occlusion_robustness.py └── print_config.py ├── deployment ├── pytorch2onnx.py ├── pytorch2torchscript.py └── test.py ├── dist_extract.sh ├── dist_test.sh ├── dist_train.sh ├── eval_tasks.sh ├── extract.py ├── kill.sh ├── model_converters ├── efficientnet_to_openmixup.py ├── extract_backbone_weights.py ├── extract_dir_weights.py ├── mlpmixer_to_openmixup.py ├── mobilenetv2_to_openmixup.py ├── publish_dir_models.py ├── publish_dir_models_mixups.py ├── publish_dir_models_timm.py ├── publish_model.py ├── pvt_to_openmixup.py ├── remove_dir.py ├── reparameterize_model.py ├── repvgg_to_openmixup.py ├── timm_to_openmixup_dir.py ├── torchvision_to_openmixup.py └── upgrade_models.py ├── prepare_data ├── convert_imagenet_subsets.py ├── convert_inaturalist.py ├── convert_subset.py ├── create_voc_data_files.py ├── create_voc_low_shot_challenge_samples.py └── prepare_voc07_cls.sh ├── srun_extract.sh ├── srun_test.sh ├── srun_train.sh ├── summary ├── find_automix_val_median.py ├── find_classification_val_median.py ├── find_regression_val_3times_average.py └── find_val_max_3times_average.py ├── test.py ├── train.py └── visualizations ├── dist_vis_fourier.sh ├── dist_vis_loss.sh ├── vis_cam.py ├── vis_fourier.py ├── vis_loss_landscape.py └── vis_lr.py /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.coveragerc -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/----.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/ISSUE_TEMPLATE/----.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/---.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/ISSUE_TEMPLATE/---.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/---bug.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/ISSUE_TEMPLATE/---bug.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/general-questions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/ISSUE_TEMPLATE/general-questions.md -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.readthedocs.yml -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/.style.yapf -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/README.md -------------------------------------------------------------------------------- /benchmarks/classification/dist_train_ft_4gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/dist_train_ft_4gpu.sh -------------------------------------------------------------------------------- /benchmarks/classification/dist_train_ft_8gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/dist_train_ft_8gpu.sh -------------------------------------------------------------------------------- /benchmarks/classification/dist_train_linear.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/dist_train_linear.sh -------------------------------------------------------------------------------- /benchmarks/classification/dist_train_linear_1gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/dist_train_linear_1gpu.sh -------------------------------------------------------------------------------- /benchmarks/classification/dist_train_linear_2gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/dist_train_linear_2gpu.sh -------------------------------------------------------------------------------- /benchmarks/classification/dist_train_linear_4gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/dist_train_linear_4gpu.sh -------------------------------------------------------------------------------- /benchmarks/classification/dist_train_semi.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/dist_train_semi.sh -------------------------------------------------------------------------------- /benchmarks/classification/knn_imagenet/knn_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/knn_imagenet/knn_classifier.py -------------------------------------------------------------------------------- /benchmarks/classification/knn_imagenet/test_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/knn_imagenet/test_knn.py -------------------------------------------------------------------------------- /benchmarks/classification/srun_train_ft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/srun_train_ft.sh -------------------------------------------------------------------------------- /benchmarks/classification/srun_train_linear.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/srun_train_linear.sh -------------------------------------------------------------------------------- /benchmarks/classification/srun_train_semi.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/srun_train_semi.sh -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/dist_extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/dist_extract.sh -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/dist_test_svm_epoch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/dist_test_svm_epoch.sh -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/extract.py -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/slurm_extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/slurm_extract.sh -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/slurm_test_svm_epoch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/slurm_test_svm_epoch.sh -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/tools/eval_svm_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/tools/eval_svm_full.sh -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/tools/svm_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/tools/svm_helper.py -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/tools/test_svm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/tools/test_svm.py -------------------------------------------------------------------------------- /benchmarks/classification/svm_voc07/tools/train_svm_kfold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/classification/svm_voc07/tools/train_svm_kfold.py -------------------------------------------------------------------------------- /benchmarks/detectron2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/detectron2/README.md -------------------------------------------------------------------------------- /benchmarks/detectron2/convert-pretrain-to-detectron2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/detectron2/convert-pretrain-to-detectron2.py -------------------------------------------------------------------------------- /benchmarks/detectron2/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/detectron2/run.sh -------------------------------------------------------------------------------- /benchmarks/detectron2/slurm_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/detectron2/slurm_run.sh -------------------------------------------------------------------------------- /benchmarks/detectron2/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/detectron2/train_net.py -------------------------------------------------------------------------------- /benchmarks/mmdetection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/README.md -------------------------------------------------------------------------------- /benchmarks/mmdetection/mim_dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/mim_dist_test.sh -------------------------------------------------------------------------------- /benchmarks/mmdetection/mim_dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/mim_dist_train.sh -------------------------------------------------------------------------------- /benchmarks/mmdetection/mim_dist_train_c4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/mim_dist_train_c4.sh -------------------------------------------------------------------------------- /benchmarks/mmdetection/mim_dist_train_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/mim_dist_train_full.sh -------------------------------------------------------------------------------- /benchmarks/mmdetection/mim_slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/mim_slurm_test.sh -------------------------------------------------------------------------------- /benchmarks/mmdetection/mim_slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/mim_slurm_train.sh -------------------------------------------------------------------------------- /benchmarks/mmdetection/mim_slurm_train_c4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/mim_slurm_train_c4.sh -------------------------------------------------------------------------------- /benchmarks/mmdetection/res_layer_extra_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmdetection/res_layer_extra_norm.py -------------------------------------------------------------------------------- /benchmarks/mmsegmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmsegmentation/README.md -------------------------------------------------------------------------------- /benchmarks/mmsegmentation/mim_dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmsegmentation/mim_dist_test.sh -------------------------------------------------------------------------------- /benchmarks/mmsegmentation/mim_dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmsegmentation/mim_dist_train.sh -------------------------------------------------------------------------------- /benchmarks/mmsegmentation/mim_dist_train_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmsegmentation/mim_dist_train_full.sh -------------------------------------------------------------------------------- /benchmarks/mmsegmentation/mim_slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmsegmentation/mim_slurm_test.sh -------------------------------------------------------------------------------- /benchmarks/mmsegmentation/mim_slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/benchmarks/mmsegmentation/mim_slurm_train.sh -------------------------------------------------------------------------------- /configs/benchmarks/classification/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/classification/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/benchmarks/classification/_base_/models/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/classification/_base_/models/r18.py -------------------------------------------------------------------------------- /configs/benchmarks/classification/_base_/models/r18_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/classification/_base_/models/r18_cifar.py -------------------------------------------------------------------------------- /configs/benchmarks/classification/_base_/models/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/classification/_base_/models/r50.py -------------------------------------------------------------------------------- /configs/benchmarks/classification/_base_/models/r50_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/classification/_base_/models/r50_cifar.py -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/Base-Keypoint-RCNN-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/Base-Keypoint-RCNN-FPN.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/Base-RCNN-C4-BN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/Base-RCNN-C4-BN.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/Base-RCNN-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/Base-RCNN-FPN.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/Base-RetinaNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/Base-RetinaNet.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_C4_1x.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_C4_1x_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_C4_1x_moco.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_C4_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_C4_2x.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_C4_2x_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_C4_2x_moco.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_FPN_1x.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_FPN_1x_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_FPN_1x_moco.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_FPN_2x.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_FPN_2x_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_FPN_2x_moco.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_RetinaNet_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_RetinaNet_1x.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/coco_R_50_RetinaNet_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/coco_R_50_RetinaNet_2x.yaml -------------------------------------------------------------------------------- /configs/benchmarks/detectron2/pascal_voc_R_50_C4_24k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/detectron2/pascal_voc_R_50_C4_24k.yaml -------------------------------------------------------------------------------- /configs/benchmarks/mmdetection/_base_/datasets/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/mmdetection/_base_/datasets/voc0712.py -------------------------------------------------------------------------------- /configs/benchmarks/mmdetection/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/benchmarks/mmdetection/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/classification/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/beit/beit_base_p16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/beit/beit_base_p16.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/davit/davit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/davit/davit_base.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/davit/davit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/davit/davit_small.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/davit/davit_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/davit/davit_tiny.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hornet/hornet_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hornet/hornet_base.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hornet/hornet_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hornet/hornet_tiny.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hrnet/hrnet_w18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hrnet/hrnet_w18.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hrnet/hrnet_w30.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hrnet/hrnet_w30.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hrnet/hrnet_w32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hrnet/hrnet_w32.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hrnet/hrnet_w40.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hrnet/hrnet_w40.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hrnet/hrnet_w44.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hrnet/hrnet_w44.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hrnet/hrnet_w48.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hrnet/hrnet_w48.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/hrnet/hrnet_w64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/hrnet/hrnet_w64.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/lit_v2/lit_v2_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/lit_v2/lit_v2_base.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/moganet/moga_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/moganet/moga_base.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/moganet/moga_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/moganet/moga_large.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/moganet/moga_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/moganet/moga_small.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/moganet/moga_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/moganet/moga_tiny.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/moganet/moga_xtiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/moganet/moga_xtiny.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/mvit/mvit_v2_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/mvit/mvit_v2_base.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/mvit/mvit_v2_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/mvit/mvit_v2_large.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/mvit/mvit_v2_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/mvit/mvit_v2_small.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/mvit/mvit_v2_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/mvit/mvit_v2_tiny.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/pvt/pvt_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/pvt/pvt_large.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/pvt/pvt_medium.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/pvt/pvt_medium.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/pvt/pvt_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/pvt/pvt_small.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/pvt/pvt_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/pvt/pvt_tiny.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/repmlp/regmlp_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/repmlp/regmlp_base.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/repvgg/repvgg_A0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/repvgg/repvgg_A0.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/repvgg/repvgg_B0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/repvgg/repvgg_B0.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/repvgg/repvgg_B3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/repvgg/repvgg_B3.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/resnet/resnet101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/resnet/resnet101.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/resnet/resnet152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/resnet/resnet152.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/resnet/resnet18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/resnet/resnet18.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/resnet/resnet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/resnet/resnet34.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/resnet/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/resnet/resnet50.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/resnet/seresnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/resnet/seresnet50.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/rwkv/rwkv_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/rwkv/rwkv_base.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/rwkv/rwkv_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/rwkv/rwkv_small.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/rwkv/rwkv_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/rwkv/rwkv_tiny.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/starnet/starnet_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/starnet/starnet_s1.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/starnet/starnet_s2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/starnet/starnet_s2.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/starnet/starnet_s3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/starnet/starnet_s3.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/starnet/starnet_s4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/starnet/starnet_s4.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/van/van_b4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/van/van_b4.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/van/van_b5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/van/van_b5.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/van/van_b6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/van/van_b6.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/van/van_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/van/van_base.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/van/van_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/van/van_large.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/van/van_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/van/van_small.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/van/van_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/van/van_tiny.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/vgg/vgg11.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/vgg/vgg11.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/vgg/vgg11_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/vgg/vgg11_bn.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/vgg/vgg13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/vgg/vgg13.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/vgg/vgg13_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/vgg/vgg13_bn.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/vgg/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/vgg/vgg16.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/vgg/vgg16_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/vgg/vgg16_bn.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/vgg/vgg19.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/vgg/vgg19.py -------------------------------------------------------------------------------- /configs/classification/_base_/models/vgg/vgg19_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/_base_/models/vgg/vgg19_bn.py -------------------------------------------------------------------------------- /configs/classification/aircrafts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/aircrafts/README.md -------------------------------------------------------------------------------- /configs/classification/cifar10/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cifar10/README.md -------------------------------------------------------------------------------- /configs/classification/cifar10/mixups/auto_train_mixups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cifar10/mixups/auto_train_mixups.py -------------------------------------------------------------------------------- /configs/classification/cifar100/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cifar100/README.md -------------------------------------------------------------------------------- /configs/classification/cifar100/wa/densenet121_CE_bs100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cifar100/wa/densenet121_CE_bs100.py -------------------------------------------------------------------------------- /configs/classification/cifar100/wa/resnet101_CE_bs100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cifar100/wa/resnet101_CE_bs100.py -------------------------------------------------------------------------------- /configs/classification/cifar100/wa/resnet18_CE_bs100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cifar100/wa/resnet18_CE_bs100.py -------------------------------------------------------------------------------- /configs/classification/cifar100/wa/resnext50_CE_bs100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cifar100/wa/resnext50_CE_bs100.py -------------------------------------------------------------------------------- /configs/classification/cifar100/wa/wrn28_10_CE_bs100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cifar100/wa/wrn28_10_CE_bs100.py -------------------------------------------------------------------------------- /configs/classification/cub200/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cub200/README.md -------------------------------------------------------------------------------- /configs/classification/cub200/mixups/auto_train_mixups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/cub200/mixups/auto_train_mixups.py -------------------------------------------------------------------------------- /configs/classification/imagenet/adautomix/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/adautomix/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/alexnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/alexnet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/automix/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/automix/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/beit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/beit/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/context_cluster/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/context_cluster/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/convmixer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/convmixer/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/convnext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/convnext/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/convnext_v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/convnext_v2/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/cspnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/cspnet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/davit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/davit/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/deit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/deit/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/deit3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/deit3/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/densenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/densenet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/edgenext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/edgenext/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/efficientformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/efficientformer/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/efficientnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/efficientnet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/efficientnet_v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/efficientnet_v2/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/hornet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/hornet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/hrnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/hrnet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/inception_v3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/inception_v3/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/levit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/levit/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/lit_v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/lit_v2/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/metaformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/metaformer/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/mixups/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/mixups/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/mlp_mixer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/mlp_mixer/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/mobilenet_v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/mobilenet_v2/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/mobilenet_v3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/mobilenet_v3/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/mobileone/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/mobileone/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/mobilevit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/mobilevit/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/moganet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/moganet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/mvit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/mvit/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/poolformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/poolformer/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/pvt/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/pvt/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/regnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/regnet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/repmlp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/repmlp/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/repvgg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/repvgg/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/res2net/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/res2net/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/resnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/resnet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/rwkv/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/rwkv/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/samix/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/samix/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/shufflenet_v1/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/shufflenet_v1/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/shufflenet_v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/shufflenet_v2/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/starnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/starnet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/swin_transformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/swin_transformer/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/t2t_vit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/t2t_vit/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/transnext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/transnext/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/twins/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/twins/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/uniformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/uniformer/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/unireplknet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/unireplknet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/van/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/van/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/van/van_b4_8xb128_ep300.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/van/van_b4_8xb128_ep300.py -------------------------------------------------------------------------------- /configs/classification/imagenet/vanillanet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/vanillanet/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/vgg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/vgg/README.md -------------------------------------------------------------------------------- /configs/classification/imagenet/vgg/vgg11_4xb64_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/vgg/vgg11_4xb64_ep100.py -------------------------------------------------------------------------------- /configs/classification/imagenet/vgg/vgg13_4xb64_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/vgg/vgg13_4xb64_ep100.py -------------------------------------------------------------------------------- /configs/classification/imagenet/vgg/vgg16_4xb64_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/vgg/vgg16_4xb64_ep100.py -------------------------------------------------------------------------------- /configs/classification/imagenet/vgg/vgg19_4xb64_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/imagenet/vgg/vgg19_4xb64_ep100.py -------------------------------------------------------------------------------- /configs/classification/inaturalist2017/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/inaturalist2017/README.md -------------------------------------------------------------------------------- /configs/classification/inaturalist2018/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/inaturalist2018/README.md -------------------------------------------------------------------------------- /configs/classification/place205/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/place205/README.md -------------------------------------------------------------------------------- /configs/classification/place205/auto_train_mixups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/place205/auto_train_mixups.py -------------------------------------------------------------------------------- /configs/classification/tiny_imagenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/classification/tiny_imagenet/README.md -------------------------------------------------------------------------------- /configs/regression/_base_/datasets/rcfmnist/sz32_bs100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/regression/_base_/datasets/rcfmnist/sz32_bs100.py -------------------------------------------------------------------------------- /configs/regression/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/regression/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/regression/_base_/models/convnext/convnext_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/regression/_base_/models/convnext/convnext_tiny.py -------------------------------------------------------------------------------- /configs/regression/_base_/models/resnet/resnet18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/regression/_base_/models/resnet/resnet18.py -------------------------------------------------------------------------------- /configs/regression/_base_/models/resnet/resnet18_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/regression/_base_/models/resnet/resnet18_cifar.py -------------------------------------------------------------------------------- /configs/regression/_base_/models/resnet/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/regression/_base_/models/resnet/resnet50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/cifar10/mae_sz224_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/cifar10/mae_sz224_bs64.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/cifar10/metrics_umap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/cifar10/metrics_umap.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/cifar100/mae_sz224_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/cifar100/mae_sz224_bs64.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/cifar100/metrics_umap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/cifar100/metrics_umap.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/imagenet/cae_sz224_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/imagenet/cae_sz224_bs64.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/imagenet/metrics_umap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/imagenet/metrics_umap.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/imagenet/odc_sz224_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/imagenet/odc_sz224_bs64.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/stl10/byol_sz96_bs256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/stl10/byol_sz96_bs256.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/stl10/mocov1_sz96_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/stl10/mocov1_sz96_bs64.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/stl10/mocov2_sz96_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/stl10/mocov2_sz96_bs64.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/stl10/simclr_sz96_bs256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/stl10/simclr_sz96_bs256.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/datasets/stl10/simmim_sz224_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/datasets/stl10/simmim_sz224_bs64.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/convnext_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/convnext_b.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/convnext_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/convnext_s.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/convnext_t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/convnext_t.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/deit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/deit_small.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/moganet_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/moganet_s.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/r101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/r101.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/r152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/r152.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/r200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/r200.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/swin_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/swin_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/swin_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/swin_tiny.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/vit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/vit_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/vit_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/vit_large.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/a2mim/vit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/a2mim/vit_small.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/barlowtwins/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/barlowtwins/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/barlowtwins/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/barlowtwins/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/beit/deit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/beit/deit_small.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/beit/vit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/beit/vit_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/byol/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/byol/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/byol/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/byol/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/cae/vit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/cae/vit_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/deepcluster/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/deepcluster/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/deepcluster/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/deepcluster/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/densecl/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/densecl/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/densecl/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/densecl/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/dino/vit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/dino/vit_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mae/vit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mae/vit_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mae/vit_huge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mae/vit_huge.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mae/vit_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mae/vit_large.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mae/vit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mae/vit_small.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/maskfeat/deit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/maskfeat/deit_small.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/maskfeat/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/maskfeat/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/maskfeat/swin_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/maskfeat/swin_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/maskfeat/vit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/maskfeat/vit_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mocov1/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mocov1/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mocov1/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mocov1/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mocov2/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mocov2/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mocov2/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mocov2/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mocov3/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mocov3/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/mocov3/vit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/mocov3/vit_small.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/npid/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/npid/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/npid/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/npid/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/odc/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/odc/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/odc/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/odc/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/relative_loc/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/relative_loc/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/relative_loc/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/relative_loc/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/rotation_pred/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/rotation_pred/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/rotation_pred/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/rotation_pred/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simclr/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simclr/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simclr/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simclr/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/convnext_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/convnext_b.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/convnext_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/convnext_s.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/convnext_t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/convnext_t.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/deit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/deit_small.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/r101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/r101.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/r152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/r152.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/r200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/r200.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/swin_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/swin_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/swin_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/swin_tiny.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/vit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/vit_base.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/vit_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/vit_large.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simmim/vit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simmim/vit_small.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simsiam/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simsiam/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/simsiam/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/simsiam/r50.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/swav/r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/swav/r18.py -------------------------------------------------------------------------------- /configs/selfsup/_base_/models/swav/r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/_base_/models/swav/r50.py -------------------------------------------------------------------------------- /configs/selfsup/a2mim/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/a2mim/README.md -------------------------------------------------------------------------------- /configs/selfsup/barlowtwins/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/barlowtwins/README.md -------------------------------------------------------------------------------- /configs/selfsup/beit/RAEDME.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/beit/RAEDME.md -------------------------------------------------------------------------------- /configs/selfsup/byol/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/byol/README.md -------------------------------------------------------------------------------- /configs/selfsup/byol/imagenet/r50_8xb64_accu8_cos_lr4_8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/byol/imagenet/r50_8xb64_accu8_cos_lr4_8.py -------------------------------------------------------------------------------- /configs/selfsup/cae/RAEDME.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/cae/RAEDME.md -------------------------------------------------------------------------------- /configs/selfsup/deepcluster/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/deepcluster/README.md -------------------------------------------------------------------------------- /configs/selfsup/densecl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/densecl/README.md -------------------------------------------------------------------------------- /configs/selfsup/densecl/cifar10/r18_4xb64_cos_ep1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/densecl/cifar10/r18_4xb64_cos_ep1000.py -------------------------------------------------------------------------------- /configs/selfsup/densecl/cifar100/r18_4xb64_cos_ep1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/densecl/cifar100/r18_4xb64_cos_ep1000.py -------------------------------------------------------------------------------- /configs/selfsup/densecl/imagenet/r18_4xb64_cos_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/densecl/imagenet/r18_4xb64_cos_ep100.py -------------------------------------------------------------------------------- /configs/selfsup/densecl/imagenet/r50_4xb64_cos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/densecl/imagenet/r50_4xb64_cos.py -------------------------------------------------------------------------------- /configs/selfsup/dino/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/dino/README.md -------------------------------------------------------------------------------- /configs/selfsup/mae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mae/README.md -------------------------------------------------------------------------------- /configs/selfsup/maskfeat/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/maskfeat/README.md -------------------------------------------------------------------------------- /configs/selfsup/mocov1/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov1/README.md -------------------------------------------------------------------------------- /configs/selfsup/mocov1/cifar10/r18_4xb64_step_ep1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov1/cifar10/r18_4xb64_step_ep1000.py -------------------------------------------------------------------------------- /configs/selfsup/mocov1/cifar100/r18_4xb64_step_ep1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov1/cifar100/r18_4xb64_step_ep1000.py -------------------------------------------------------------------------------- /configs/selfsup/mocov1/imagenet/r18_4xb64_step_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov1/imagenet/r18_4xb64_step_ep100.py -------------------------------------------------------------------------------- /configs/selfsup/mocov1/imagenet/r50_4xb64_step_ep200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov1/imagenet/r50_4xb64_step_ep200.py -------------------------------------------------------------------------------- /configs/selfsup/mocov2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov2/README.md -------------------------------------------------------------------------------- /configs/selfsup/mocov2/cifar10/r18_4xb64_cos_ep1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov2/cifar10/r18_4xb64_cos_ep1000.py -------------------------------------------------------------------------------- /configs/selfsup/mocov2/cifar100/r18_4xb64_cos_ep1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov2/cifar100/r18_4xb64_cos_ep1000.py -------------------------------------------------------------------------------- /configs/selfsup/mocov2/imagenet/r18_4xb64_cos_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov2/imagenet/r18_4xb64_cos_ep100.py -------------------------------------------------------------------------------- /configs/selfsup/mocov2/imagenet/r50_4xb64_cos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov2/imagenet/r50_4xb64_cos.py -------------------------------------------------------------------------------- /configs/selfsup/mocov2/stl10/r50_4xb64_cos_ep1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov2/stl10/r50_4xb64_cos_ep1000.py -------------------------------------------------------------------------------- /configs/selfsup/mocov3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/mocov3/README.md -------------------------------------------------------------------------------- /configs/selfsup/npid/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/npid/README.md -------------------------------------------------------------------------------- /configs/selfsup/npid/imagenet/r18_4xb64_step_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/npid/imagenet/r18_4xb64_step_ep100.py -------------------------------------------------------------------------------- /configs/selfsup/npid/imagenet/r50_4xb64_step_ep200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/npid/imagenet/r50_4xb64_step_ep200.py -------------------------------------------------------------------------------- /configs/selfsup/odc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/odc/README.md -------------------------------------------------------------------------------- /configs/selfsup/odc/imagenet/r18_8xb64_step_ep100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/odc/imagenet/r18_8xb64_step_ep100.py -------------------------------------------------------------------------------- /configs/selfsup/odc/imagenet/r50_8xb64_step_ep440.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/odc/imagenet/r50_8xb64_step_ep440.py -------------------------------------------------------------------------------- /configs/selfsup/relative_loc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/relative_loc/README.md -------------------------------------------------------------------------------- /configs/selfsup/rotation_pred/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/rotation_pred/README.md -------------------------------------------------------------------------------- /configs/selfsup/simclr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/simclr/README.md -------------------------------------------------------------------------------- /configs/selfsup/simclr/imagenet/r50_8xb64_cos_lr0_6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/simclr/imagenet/r50_8xb64_cos_lr0_6.py -------------------------------------------------------------------------------- /configs/selfsup/simmim/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/simmim/README.md -------------------------------------------------------------------------------- /configs/selfsup/simsiam/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/simsiam/README.md -------------------------------------------------------------------------------- /configs/selfsup/simsiam/imagenet/r50_4xb64_cos_lr0_05.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/simsiam/imagenet/r50_4xb64_cos_lr0_05.py -------------------------------------------------------------------------------- /configs/selfsup/simsiam/imagenet/r50_8xb64_cos_lr0_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/simsiam/imagenet/r50_8xb64_cos_lr0_1.py -------------------------------------------------------------------------------- /configs/selfsup/swav/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/selfsup/swav/README.md -------------------------------------------------------------------------------- /configs/semisup/_base_/datasets/cars/per_15_sz224_bs24.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/semisup/_base_/datasets/cars/per_15_sz224_bs24.py -------------------------------------------------------------------------------- /configs/semisup/_base_/datasets/cars/per_30_sz224_bs24.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/semisup/_base_/datasets/cars/per_30_sz224_bs24.py -------------------------------------------------------------------------------- /configs/semisup/_base_/datasets/cars/per_50_sz224_bs24.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/semisup/_base_/datasets/cars/per_50_sz224_bs24.py -------------------------------------------------------------------------------- /configs/semisup/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/semisup/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/semisup/selftuning/auto_train_self_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/configs/semisup/selftuning/auto_train_self_tuning.py -------------------------------------------------------------------------------- /demo/bird.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/demo/bird.JPEG -------------------------------------------------------------------------------- /demo/cat-dog.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/demo/cat-dog.png -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docs/en/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/Makefile -------------------------------------------------------------------------------- /docs/en/_static/css/readthedocs.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/_static/css/readthedocs.css -------------------------------------------------------------------------------- /docs/en/_static/image/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/_static/image/logo.png -------------------------------------------------------------------------------- /docs/en/_templates/404.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/_templates/404.html -------------------------------------------------------------------------------- /docs/en/_templates/autosummary/class.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/_templates/autosummary/class.rst -------------------------------------------------------------------------------- /docs/en/_templates/callable.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/_templates/callable.rst -------------------------------------------------------------------------------- /docs/en/_templates/data_transform.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/_templates/data_transform.rst -------------------------------------------------------------------------------- /docs/en/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/api.rst -------------------------------------------------------------------------------- /docs/en/api/apis.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/api/apis.rst -------------------------------------------------------------------------------- /docs/en/api/core.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/api/core.rst -------------------------------------------------------------------------------- /docs/en/api/datasets.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/api/datasets.rst -------------------------------------------------------------------------------- /docs/en/api/models.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/api/models.rst -------------------------------------------------------------------------------- /docs/en/api/models.utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/api/models.utils.rst -------------------------------------------------------------------------------- /docs/en/api/third_party.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/api/third_party.rst -------------------------------------------------------------------------------- /docs/en/api/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/api/utils.rst -------------------------------------------------------------------------------- /docs/en/awesome/awesome_mixup.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/awesome/awesome_mixup.md -------------------------------------------------------------------------------- /docs/en/awesome/awesome_selfsup.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/awesome/awesome_selfsup.md -------------------------------------------------------------------------------- /docs/en/changelog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/changelog.md -------------------------------------------------------------------------------- /docs/en/compatibility.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/compatibility.md -------------------------------------------------------------------------------- /docs/en/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/conf.py -------------------------------------------------------------------------------- /docs/en/faq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/faq.md -------------------------------------------------------------------------------- /docs/en/get_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/get_started.md -------------------------------------------------------------------------------- /docs/en/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/index.rst -------------------------------------------------------------------------------- /docs/en/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/install.md -------------------------------------------------------------------------------- /docs/en/mixup_benchmarks/Mixup_cifar.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/mixup_benchmarks/Mixup_cifar.md -------------------------------------------------------------------------------- /docs/en/mixup_benchmarks/Mixup_downstream.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/mixup_benchmarks/Mixup_downstream.md -------------------------------------------------------------------------------- /docs/en/mixup_benchmarks/Mixup_imagenet.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/mixup_benchmarks/Mixup_imagenet.md -------------------------------------------------------------------------------- /docs/en/model_zoos/Mixup_sup.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/model_zoos/Mixup_sup.md -------------------------------------------------------------------------------- /docs/en/model_zoos/Model_Zoo_selfsup.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/model_zoos/Model_Zoo_selfsup.md -------------------------------------------------------------------------------- /docs/en/model_zoos/Model_Zoo_semisup.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/model_zoos/Model_Zoo_semisup.md -------------------------------------------------------------------------------- /docs/en/model_zoos/Model_Zoo_sup.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/model_zoos/Model_Zoo_sup.md -------------------------------------------------------------------------------- /docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | -------------------------------------------------------------------------------- /docs/en/tools/analysis.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tools/analysis.md -------------------------------------------------------------------------------- /docs/en/tools/pytorch2onnx.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tools/pytorch2onnx.md -------------------------------------------------------------------------------- /docs/en/tools/pytorch2torchscript.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tools/pytorch2torchscript.md -------------------------------------------------------------------------------- /docs/en/tools/visualization.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tools/visualization.md -------------------------------------------------------------------------------- /docs/en/tutorials/0_config.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/0_config.md -------------------------------------------------------------------------------- /docs/en/tutorials/1_new_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/1_new_dataset.md -------------------------------------------------------------------------------- /docs/en/tutorials/2_data_pipeline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/2_data_pipeline.md -------------------------------------------------------------------------------- /docs/en/tutorials/3_new_modules.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/3_new_modules.md -------------------------------------------------------------------------------- /docs/en/tutorials/4_schedule.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/4_schedule.md -------------------------------------------------------------------------------- /docs/en/tutorials/5_runtime.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/5_runtime.md -------------------------------------------------------------------------------- /docs/en/tutorials/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/index.rst -------------------------------------------------------------------------------- /docs/en/tutorials/ssl_classification.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/ssl_classification.md -------------------------------------------------------------------------------- /docs/en/tutorials/ssl_detection.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/ssl_detection.md -------------------------------------------------------------------------------- /docs/en/tutorials/ssl_segmentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/docs/en/tutorials/ssl_segmentation.md -------------------------------------------------------------------------------- /openmixup/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/__init__.py -------------------------------------------------------------------------------- /openmixup/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/apis/__init__.py -------------------------------------------------------------------------------- /openmixup/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/apis/test.py -------------------------------------------------------------------------------- /openmixup/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/apis/train.py -------------------------------------------------------------------------------- /openmixup/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/__init__.py -------------------------------------------------------------------------------- /openmixup/core/export/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/export/__init__.py -------------------------------------------------------------------------------- /openmixup/core/export/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/export/test.py -------------------------------------------------------------------------------- /openmixup/core/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/__init__.py -------------------------------------------------------------------------------- /openmixup/core/hooks/addtional_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/addtional_scheduler.py -------------------------------------------------------------------------------- /openmixup/core/hooks/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/builder.py -------------------------------------------------------------------------------- /openmixup/core/hooks/deepcluster_automix_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/deepcluster_automix_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/deepcluster_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/deepcluster_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/ema_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/ema_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/extractor.py -------------------------------------------------------------------------------- /openmixup/core/hooks/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/lr_scheduler.py -------------------------------------------------------------------------------- /openmixup/core/hooks/momentum_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/momentum_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/odc_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/odc_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/optimizer_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/optimizer_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/precise_bn_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/precise_bn_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/registry.py -------------------------------------------------------------------------------- /openmixup/core/hooks/save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/save_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/selfsup_metric_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/selfsup_metric_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/swav_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/swav_hook.py -------------------------------------------------------------------------------- /openmixup/core/hooks/validate_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/hooks/validate_hook.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/__init__.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/adabelief.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/adabelief.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/adabound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/adabound.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/adafactor.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/adahessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/adahessian.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/adamp.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/adan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/adan.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/builder.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/constructor.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/lamb.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/lars.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/lion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/lion.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/madgrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/madgrad.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/nvnovograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/nvnovograd.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/sam.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/sgdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/sgdp.py -------------------------------------------------------------------------------- /openmixup/core/optimizers/sophia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/core/optimizers/sophia.py -------------------------------------------------------------------------------- /openmixup/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/__init__.py -------------------------------------------------------------------------------- /openmixup/datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/base.py -------------------------------------------------------------------------------- /openmixup/datasets/beit_mim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/beit_mim.py -------------------------------------------------------------------------------- /openmixup/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/builder.py -------------------------------------------------------------------------------- /openmixup/datasets/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/classification.py -------------------------------------------------------------------------------- /openmixup/datasets/contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/contrastive.py -------------------------------------------------------------------------------- /openmixup/datasets/data_sources/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/data_sources/__init__.py -------------------------------------------------------------------------------- /openmixup/datasets/data_sources/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/data_sources/cifar.py -------------------------------------------------------------------------------- /openmixup/datasets/data_sources/image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/data_sources/image_list.py -------------------------------------------------------------------------------- /openmixup/datasets/data_sources/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/data_sources/imagenet.py -------------------------------------------------------------------------------- /openmixup/datasets/data_sources/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/data_sources/mnist.py -------------------------------------------------------------------------------- /openmixup/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /openmixup/datasets/deepcluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/deepcluster.py -------------------------------------------------------------------------------- /openmixup/datasets/deepcluster_contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/deepcluster_contrastive.py -------------------------------------------------------------------------------- /openmixup/datasets/extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/extraction.py -------------------------------------------------------------------------------- /openmixup/datasets/loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/loader/__init__.py -------------------------------------------------------------------------------- /openmixup/datasets/loader/build_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/loader/build_loader.py -------------------------------------------------------------------------------- /openmixup/datasets/loader/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/loader/sampler.py -------------------------------------------------------------------------------- /openmixup/datasets/masked_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/masked_image.py -------------------------------------------------------------------------------- /openmixup/datasets/multi_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/multi_view.py -------------------------------------------------------------------------------- /openmixup/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /openmixup/datasets/pipelines/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/pipelines/auto_augment.py -------------------------------------------------------------------------------- /openmixup/datasets/pipelines/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/pipelines/compose.py -------------------------------------------------------------------------------- /openmixup/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /openmixup/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/registry.py -------------------------------------------------------------------------------- /openmixup/datasets/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/regression.py -------------------------------------------------------------------------------- /openmixup/datasets/relative_loc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/relative_loc.py -------------------------------------------------------------------------------- /openmixup/datasets/rotation_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/rotation_pred.py -------------------------------------------------------------------------------- /openmixup/datasets/semi_supervised.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/semi_supervised.py -------------------------------------------------------------------------------- /openmixup/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/datasets/utils.py -------------------------------------------------------------------------------- /openmixup/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/__init__.py -------------------------------------------------------------------------------- /openmixup/models/augments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/__init__.py -------------------------------------------------------------------------------- /openmixup/models/augments/alignmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/alignmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/attentivemix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/attentivemix.py -------------------------------------------------------------------------------- /openmixup/models/augments/augmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/augmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/cutmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/cutmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/fmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/fmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/gridmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/gridmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/guidedmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/guidedmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/mixpro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/mixpro.py -------------------------------------------------------------------------------- /openmixup/models/augments/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/mixup.py -------------------------------------------------------------------------------- /openmixup/models/augments/puzzlemix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/puzzlemix.py -------------------------------------------------------------------------------- /openmixup/models/augments/resizemix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/resizemix.py -------------------------------------------------------------------------------- /openmixup/models/augments/saliencymix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/saliencymix.py -------------------------------------------------------------------------------- /openmixup/models/augments/smmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/smmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/smoothmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/smoothmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/snapmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/snapmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/tla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/tla.py -------------------------------------------------------------------------------- /openmixup/models/augments/tokenmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/tokenmix.py -------------------------------------------------------------------------------- /openmixup/models/augments/transmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/augments/transmix.py -------------------------------------------------------------------------------- /openmixup/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/__init__.py -------------------------------------------------------------------------------- /openmixup/models/backbones/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/alexnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/base_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/base_backbone.py -------------------------------------------------------------------------------- /openmixup/models/backbones/beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/beit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/context_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/context_cluster.py -------------------------------------------------------------------------------- /openmixup/models/backbones/convmixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/convmixer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/convnext.py -------------------------------------------------------------------------------- /openmixup/models/backbones/cspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/cspnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/davit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/davit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/deit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/deit3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/deit3.py -------------------------------------------------------------------------------- /openmixup/models/backbones/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/densenet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/edgenext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/edgenext.py -------------------------------------------------------------------------------- /openmixup/models/backbones/efficientformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/efficientformer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/efficientnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/efficientnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/efficientnet_v2.py -------------------------------------------------------------------------------- /openmixup/models/backbones/hornet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/hornet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/hrnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/inception_v3.py -------------------------------------------------------------------------------- /openmixup/models/backbones/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/lenet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/levit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/levit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/lit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/lit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/metaformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/metaformer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mim_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mim_resnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mim_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mim_swin.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mim_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mim_vit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mlp_mixer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mobilenet_v2.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mobilenet_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mobilenet_v3.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mobileone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mobileone.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mobilevit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mobilevit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/moganet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/moganet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/mvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/mvit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/poolformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/poolformer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/pvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/pvt.py -------------------------------------------------------------------------------- /openmixup/models/backbones/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/regnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/replknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/replknet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/repmlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/repmlp.py -------------------------------------------------------------------------------- /openmixup/models/backbones/repvgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/repvgg.py -------------------------------------------------------------------------------- /openmixup/models/backbones/res2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/res2net.py -------------------------------------------------------------------------------- /openmixup/models/backbones/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/resnest.py -------------------------------------------------------------------------------- /openmixup/models/backbones/resnet_mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/resnet_mmcls.py -------------------------------------------------------------------------------- /openmixup/models/backbones/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/resnext.py -------------------------------------------------------------------------------- /openmixup/models/backbones/revvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/revvit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/riformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/riformer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/rwkv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/rwkv.py -------------------------------------------------------------------------------- /openmixup/models/backbones/seresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/seresnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/shufflenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/shufflenet_v1.py -------------------------------------------------------------------------------- /openmixup/models/backbones/shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/shufflenet_v2.py -------------------------------------------------------------------------------- /openmixup/models/backbones/starnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/starnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/swin_transformer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/swin_transformer_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/swin_transformer_v2.py -------------------------------------------------------------------------------- /openmixup/models/backbones/t2t_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/t2t_vit.py -------------------------------------------------------------------------------- /openmixup/models/backbones/timm_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/timm_backbone.py -------------------------------------------------------------------------------- /openmixup/models/backbones/transnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/transnext.py -------------------------------------------------------------------------------- /openmixup/models/backbones/twins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/twins.py -------------------------------------------------------------------------------- /openmixup/models/backbones/uniformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/uniformer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/unireplknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/unireplknet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/van.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/van.py -------------------------------------------------------------------------------- /openmixup/models/backbones/vanillanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/vanillanet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/vgg.py -------------------------------------------------------------------------------- /openmixup/models/backbones/vig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/vig.py -------------------------------------------------------------------------------- /openmixup/models/backbones/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/vision_transformer.py -------------------------------------------------------------------------------- /openmixup/models/backbones/wide_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/wide_resnet.py -------------------------------------------------------------------------------- /openmixup/models/backbones/xcit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/backbones/xcit.py -------------------------------------------------------------------------------- /openmixup/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/builder.py -------------------------------------------------------------------------------- /openmixup/models/classifiers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/classifiers/__init__.py -------------------------------------------------------------------------------- /openmixup/models/classifiers/adautomix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/classifiers/adautomix.py -------------------------------------------------------------------------------- /openmixup/models/classifiers/automix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/classifiers/automix.py -------------------------------------------------------------------------------- /openmixup/models/classifiers/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/classifiers/base_model.py -------------------------------------------------------------------------------- /openmixup/models/classifiers/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/classifiers/classification.py -------------------------------------------------------------------------------- /openmixup/models/classifiers/mergemix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/classifiers/mergemix.py -------------------------------------------------------------------------------- /openmixup/models/classifiers/mixup_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/classifiers/mixup_classification.py -------------------------------------------------------------------------------- /openmixup/models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/__init__.py -------------------------------------------------------------------------------- /openmixup/models/heads/adaptive_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/adaptive_mask.py -------------------------------------------------------------------------------- /openmixup/models/heads/cls_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/cls_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/cls_mixup_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/cls_mixup_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/cls_mlp_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/cls_mlp_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/contrastive_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/contrastive_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/dino_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/dino_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/latent_pred_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/latent_pred_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/mim_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/mim_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/multi_cls_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/multi_cls_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/norm_linear_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/norm_linear_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/pmix_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/pmix_block.py -------------------------------------------------------------------------------- /openmixup/models/heads/reg_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/reg_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/swav_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/swav_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/tokenizer_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/tokenizer_head.py -------------------------------------------------------------------------------- /openmixup/models/heads/vision_transformer_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/heads/vision_transformer_head.py -------------------------------------------------------------------------------- /openmixup/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/__init__.py -------------------------------------------------------------------------------- /openmixup/models/losses/arcface_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/arcface_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/asymmetric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/asymmetric_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/distillation_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/distillation_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/focal_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/label_smooth_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/label_smooth_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/regression_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/regression_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/seesaw_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/seesaw_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/semantic_softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/semantic_softmax_loss.py -------------------------------------------------------------------------------- /openmixup/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/losses/utils.py -------------------------------------------------------------------------------- /openmixup/models/memories/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/memories/__init__.py -------------------------------------------------------------------------------- /openmixup/models/memories/odc_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/memories/odc_memory.py -------------------------------------------------------------------------------- /openmixup/models/memories/simple_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/memories/simple_memory.py -------------------------------------------------------------------------------- /openmixup/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/necks/__init__.py -------------------------------------------------------------------------------- /openmixup/models/necks/conv_necks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/necks/conv_necks.py -------------------------------------------------------------------------------- /openmixup/models/necks/hr_necks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/necks/hr_necks.py -------------------------------------------------------------------------------- /openmixup/models/necks/mim_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/necks/mim_neck.py -------------------------------------------------------------------------------- /openmixup/models/necks/mlp_necks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/necks/mlp_necks.py -------------------------------------------------------------------------------- /openmixup/models/necks/transformer_necks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/necks/transformer_necks.py -------------------------------------------------------------------------------- /openmixup/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/registry.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/__init__.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/a2mim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/a2mim.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/barlowtwins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/barlowtwins.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/beit.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/byol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/byol.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/cae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/cae.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/deepcluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/deepcluster.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/densecl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/densecl.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/dino.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/mae.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/maskfeat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/maskfeat.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/moco.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/moco_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/moco_mix.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/moco_samix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/moco_samix.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/mocov3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/mocov3.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/npid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/npid.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/odc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/odc.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/relative_loc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/relative_loc.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/rotation_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/rotation_pred.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/simclr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/simclr.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/simclr_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/simclr_mix.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/simmim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/simmim.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/simsiam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/simsiam.py -------------------------------------------------------------------------------- /openmixup/models/selfsup/swav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/selfsup/swav.py -------------------------------------------------------------------------------- /openmixup/models/semisup/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/semisup/__init__.py -------------------------------------------------------------------------------- /openmixup/models/semisup/dmix_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/semisup/dmix_tuning.py -------------------------------------------------------------------------------- /openmixup/models/semisup/dmixmatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/semisup/dmixmatch.py -------------------------------------------------------------------------------- /openmixup/models/semisup/fine_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/semisup/fine_tuning.py -------------------------------------------------------------------------------- /openmixup/models/semisup/fixmatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/semisup/fixmatch.py -------------------------------------------------------------------------------- /openmixup/models/semisup/self_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/semisup/self_tuning.py -------------------------------------------------------------------------------- /openmixup/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/__init__.py -------------------------------------------------------------------------------- /openmixup/models/utils/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/accuracy.py -------------------------------------------------------------------------------- /openmixup/models/utils/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/augmentation.py -------------------------------------------------------------------------------- /openmixup/models/utils/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/evaluation/__init__.py -------------------------------------------------------------------------------- /openmixup/models/utils/evaluation/eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/evaluation/eval_metrics.py -------------------------------------------------------------------------------- /openmixup/models/utils/evaluation/mean_ap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/evaluation/mean_ap.py -------------------------------------------------------------------------------- /openmixup/models/utils/gather_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/gather_layer.py -------------------------------------------------------------------------------- /openmixup/models/utils/grad_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/grad_weight.py -------------------------------------------------------------------------------- /openmixup/models/utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/helpers.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/__init__.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/attention.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/channel_shuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/channel_shuffle.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/conv_ws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/conv_ws.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/dall_e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/dall_e.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/drop.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/inverted_residual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/inverted_residual.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/layer_scale.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/make_divisible.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/make_divisible.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/norm.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/patch_embed.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/pooling.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/pos_embed.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/res_layer_extra_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/res_layer_extra_norm.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/scale.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/se_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/se_layer.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/smoothing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/smoothing.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/sobel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/sobel.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/sparse_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/sparse_modules.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/transformer.py -------------------------------------------------------------------------------- /openmixup/models/utils/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/layers/weight_init.py -------------------------------------------------------------------------------- /openmixup/models/utils/merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/merge.py -------------------------------------------------------------------------------- /openmixup/models/utils/target_generators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/target_generators/__init__.py -------------------------------------------------------------------------------- /openmixup/models/utils/target_generators/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/target_generators/clip.py -------------------------------------------------------------------------------- /openmixup/models/utils/target_generators/clip_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/target_generators/clip_generator.py -------------------------------------------------------------------------------- /openmixup/models/utils/target_generators/dall_e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/target_generators/dall_e.py -------------------------------------------------------------------------------- /openmixup/models/utils/target_generators/hog_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/target_generators/hog_generator.py -------------------------------------------------------------------------------- /openmixup/models/utils/target_generators/vqkd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/target_generators/vqkd.py -------------------------------------------------------------------------------- /openmixup/models/utils/visualization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/visualization/__init__.py -------------------------------------------------------------------------------- /openmixup/models/utils/visualization/draw_hog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/visualization/draw_hog.py -------------------------------------------------------------------------------- /openmixup/models/utils/visualization/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/visualization/image.py -------------------------------------------------------------------------------- /openmixup/models/utils/visualization/plot_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/models/utils/visualization/plot_torch.py -------------------------------------------------------------------------------- /openmixup/third_party/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/third_party/__init__.py -------------------------------------------------------------------------------- /openmixup/third_party/clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/third_party/clustering.py -------------------------------------------------------------------------------- /openmixup/third_party/distributed_sinkhorn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/third_party/distributed_sinkhorn.py -------------------------------------------------------------------------------- /openmixup/third_party/knn_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/third_party/knn_classifier.py -------------------------------------------------------------------------------- /openmixup/third_party/svm_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/third_party/svm_classifier.py -------------------------------------------------------------------------------- /openmixup/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/__init__.py -------------------------------------------------------------------------------- /openmixup/utils/alias_multinomial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/alias_multinomial.py -------------------------------------------------------------------------------- /openmixup/utils/collect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/collect.py -------------------------------------------------------------------------------- /openmixup/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/collect_env.py -------------------------------------------------------------------------------- /openmixup/utils/config_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/config_tools.py -------------------------------------------------------------------------------- /openmixup/utils/contextmanagers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/contextmanagers.py -------------------------------------------------------------------------------- /openmixup/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/dist_utils.py -------------------------------------------------------------------------------- /openmixup/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/flops_counter.py -------------------------------------------------------------------------------- /openmixup/utils/fp16_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/fp16_utils.py -------------------------------------------------------------------------------- /openmixup/utils/gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/gather.py -------------------------------------------------------------------------------- /openmixup/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/logger.py -------------------------------------------------------------------------------- /openmixup/utils/loss_landscape_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/loss_landscape_utils.py -------------------------------------------------------------------------------- /openmixup/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/misc.py -------------------------------------------------------------------------------- /openmixup/utils/profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/profiling.py -------------------------------------------------------------------------------- /openmixup/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/registry.py -------------------------------------------------------------------------------- /openmixup/utils/setup_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/utils/setup_env.py -------------------------------------------------------------------------------- /openmixup/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/openmixup/version.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/requirements/docs.txt -------------------------------------------------------------------------------- /requirements/mminstall.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/requirements/mminstall.txt -------------------------------------------------------------------------------- /requirements/optional.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/requirements/optional.txt -------------------------------------------------------------------------------- /requirements/readthedocs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/requirements/readthedocs.txt -------------------------------------------------------------------------------- /requirements/runtime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/requirements/runtime.txt -------------------------------------------------------------------------------- /requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/requirements/tests.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_datasets/test_dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_datasets/test_dataset_utils.py -------------------------------------------------------------------------------- /tests/test_datasets/test_dataset_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_datasets/test_dataset_vis.py -------------------------------------------------------------------------------- /tests/test_datasets/test_pipelines/test_auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_datasets/test_pipelines/test_auto_augment.py -------------------------------------------------------------------------------- /tests/test_datasets/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_datasets/test_utils.py -------------------------------------------------------------------------------- /tests/test_metrics/test_eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_metrics/test_eval_metrics.py -------------------------------------------------------------------------------- /tests/test_metrics/test_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_metrics/test_loss.py -------------------------------------------------------------------------------- /tests/test_metrics/test_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_metrics/test_tools.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_convmixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_convmixer.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_convnext.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_deit.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_densenet.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_edgenext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_edgenext.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_efficientformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_efficientformer.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_efficientnet.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_hornet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_hornet.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_mlp_mixer.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_mobilenet_v2.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_mobilenet_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_mobilenet_v3.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_mvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_mvit.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_poolformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_poolformer.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_resnest.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_resnet_mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_resnet_mmcls.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_resnext.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_seresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_seresnet.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_shufflenet_v1.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_shufflenet_v2.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_timm_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_timm_backbone.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_van.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_van.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/test_vgg.py -------------------------------------------------------------------------------- /tests/test_models/test_backbones/test_vision_transformer.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_models/test_backbones/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_backbones/utils.py -------------------------------------------------------------------------------- /tests/test_models/test_classifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_classifiers.py -------------------------------------------------------------------------------- /tests/test_models/test_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_heads.py -------------------------------------------------------------------------------- /tests/test_models/test_necks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_necks.py -------------------------------------------------------------------------------- /tests/test_models/test_utils/test_augments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_utils/test_augments.py -------------------------------------------------------------------------------- /tests/test_models/test_utils/test_layers/test_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_utils/test_layers/test_attention.py -------------------------------------------------------------------------------- /tests/test_models/test_utils/test_layers/test_pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_utils/test_layers/test_pos_embed.py -------------------------------------------------------------------------------- /tests/test_models/test_utils/test_layers/test_se_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_models/test_utils/test_layers/test_se_layer.py -------------------------------------------------------------------------------- /tests/test_runtime/test_eval_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_runtime/test_eval_hook.py -------------------------------------------------------------------------------- /tests/test_runtime/test_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_runtime/test_hooks.py -------------------------------------------------------------------------------- /tests/test_runtime/test_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_runtime/test_optimizer.py -------------------------------------------------------------------------------- /tests/test_runtime/test_preciseBN_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_runtime/test_preciseBN_hook.py -------------------------------------------------------------------------------- /tests/test_utils/test_setup_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_utils/test_setup_env.py -------------------------------------------------------------------------------- /tests/test_utils/test_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tests/test_utils/test_visualization.py -------------------------------------------------------------------------------- /tools/analysis_tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/analysis_tools/analyze_logs.py -------------------------------------------------------------------------------- /tools/analysis_tools/calibration_fgsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/analysis_tools/calibration_fgsm.py -------------------------------------------------------------------------------- /tools/analysis_tools/count_parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/analysis_tools/count_parameters.py -------------------------------------------------------------------------------- /tools/analysis_tools/courrption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/analysis_tools/courrption.py -------------------------------------------------------------------------------- /tools/analysis_tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/analysis_tools/get_flops.py -------------------------------------------------------------------------------- /tools/analysis_tools/merge_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/analysis_tools/merge_logs.py -------------------------------------------------------------------------------- /tools/analysis_tools/occlusion_robustness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/analysis_tools/occlusion_robustness.py -------------------------------------------------------------------------------- /tools/analysis_tools/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/analysis_tools/print_config.py -------------------------------------------------------------------------------- /tools/deployment/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/deployment/pytorch2onnx.py -------------------------------------------------------------------------------- /tools/deployment/pytorch2torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/deployment/pytorch2torchscript.py -------------------------------------------------------------------------------- /tools/deployment/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/deployment/test.py -------------------------------------------------------------------------------- /tools/dist_extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/dist_extract.sh -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/eval_tasks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/eval_tasks.sh -------------------------------------------------------------------------------- /tools/extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/extract.py -------------------------------------------------------------------------------- /tools/kill.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/kill.sh -------------------------------------------------------------------------------- /tools/model_converters/efficientnet_to_openmixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/efficientnet_to_openmixup.py -------------------------------------------------------------------------------- /tools/model_converters/extract_backbone_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/extract_backbone_weights.py -------------------------------------------------------------------------------- /tools/model_converters/extract_dir_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/extract_dir_weights.py -------------------------------------------------------------------------------- /tools/model_converters/mlpmixer_to_openmixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/mlpmixer_to_openmixup.py -------------------------------------------------------------------------------- /tools/model_converters/mobilenetv2_to_openmixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/mobilenetv2_to_openmixup.py -------------------------------------------------------------------------------- /tools/model_converters/publish_dir_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/publish_dir_models.py -------------------------------------------------------------------------------- /tools/model_converters/publish_dir_models_mixups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/publish_dir_models_mixups.py -------------------------------------------------------------------------------- /tools/model_converters/publish_dir_models_timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/publish_dir_models_timm.py -------------------------------------------------------------------------------- /tools/model_converters/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/publish_model.py -------------------------------------------------------------------------------- /tools/model_converters/pvt_to_openmixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/pvt_to_openmixup.py -------------------------------------------------------------------------------- /tools/model_converters/remove_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/remove_dir.py -------------------------------------------------------------------------------- /tools/model_converters/reparameterize_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/reparameterize_model.py -------------------------------------------------------------------------------- /tools/model_converters/repvgg_to_openmixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/repvgg_to_openmixup.py -------------------------------------------------------------------------------- /tools/model_converters/timm_to_openmixup_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/timm_to_openmixup_dir.py -------------------------------------------------------------------------------- /tools/model_converters/torchvision_to_openmixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/torchvision_to_openmixup.py -------------------------------------------------------------------------------- /tools/model_converters/upgrade_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/model_converters/upgrade_models.py -------------------------------------------------------------------------------- /tools/prepare_data/convert_imagenet_subsets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/prepare_data/convert_imagenet_subsets.py -------------------------------------------------------------------------------- /tools/prepare_data/convert_inaturalist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/prepare_data/convert_inaturalist.py -------------------------------------------------------------------------------- /tools/prepare_data/convert_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/prepare_data/convert_subset.py -------------------------------------------------------------------------------- /tools/prepare_data/create_voc_data_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/prepare_data/create_voc_data_files.py -------------------------------------------------------------------------------- /tools/prepare_data/prepare_voc07_cls.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/prepare_data/prepare_voc07_cls.sh -------------------------------------------------------------------------------- /tools/srun_extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/srun_extract.sh -------------------------------------------------------------------------------- /tools/srun_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/srun_test.sh -------------------------------------------------------------------------------- /tools/srun_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/srun_train.sh -------------------------------------------------------------------------------- /tools/summary/find_automix_val_median.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/summary/find_automix_val_median.py -------------------------------------------------------------------------------- /tools/summary/find_classification_val_median.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/summary/find_classification_val_median.py -------------------------------------------------------------------------------- /tools/summary/find_regression_val_3times_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/summary/find_regression_val_3times_average.py -------------------------------------------------------------------------------- /tools/summary/find_val_max_3times_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/summary/find_val_max_3times_average.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/visualizations/dist_vis_fourier.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/visualizations/dist_vis_fourier.sh -------------------------------------------------------------------------------- /tools/visualizations/dist_vis_loss.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/visualizations/dist_vis_loss.sh -------------------------------------------------------------------------------- /tools/visualizations/vis_cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/visualizations/vis_cam.py -------------------------------------------------------------------------------- /tools/visualizations/vis_fourier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/visualizations/vis_fourier.py -------------------------------------------------------------------------------- /tools/visualizations/vis_loss_landscape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/visualizations/vis_loss_landscape.py -------------------------------------------------------------------------------- /tools/visualizations/vis_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Westlake-AI/openmixup/HEAD/tools/visualizations/vis_lr.py --------------------------------------------------------------------------------