gitextract_gk47a2w1/ ├── .gitignore ├── CODE_OF_CONDUCT.md ├── LICENSE ├── MODELHUB.md ├── README.md ├── SECURITY.md ├── SUPPORT.md ├── config.py ├── configs/ │ ├── simmim/ │ │ ├── simmim_finetune__swin_base__img224_window7__800ep.yaml │ │ ├── simmim_finetune__swinv2_base__img224_window14__800ep.yaml │ │ ├── simmim_pretrain__swin_base__img192_window6__800ep.yaml │ │ └── simmim_pretrain__swinv2_base__img192_window12__800ep.yaml │ ├── swin/ │ │ ├── swin_base_patch4_window12_384_22kto1k_finetune.yaml │ │ ├── swin_base_patch4_window12_384_finetune.yaml │ │ ├── swin_base_patch4_window7_224.yaml │ │ ├── swin_base_patch4_window7_224_22k.yaml │ │ ├── swin_base_patch4_window7_224_22kto1k_finetune.yaml │ │ ├── swin_large_patch4_window12_384_22kto1k_finetune.yaml │ │ ├── swin_large_patch4_window7_224_22k.yaml │ │ ├── swin_large_patch4_window7_224_22kto1k_finetune.yaml │ │ ├── swin_small_patch4_window7_224.yaml │ │ ├── swin_small_patch4_window7_224_22k.yaml │ │ ├── swin_small_patch4_window7_224_22kto1k_finetune.yaml │ │ ├── swin_tiny_c24_patch4_window8_256.yaml │ │ ├── swin_tiny_patch4_window7_224.yaml │ │ ├── swin_tiny_patch4_window7_224_22k.yaml │ │ └── swin_tiny_patch4_window7_224_22kto1k_finetune.yaml │ ├── swinmlp/ │ │ ├── swin_mlp_base_patch4_window7_224.yaml │ │ ├── swin_mlp_tiny_c12_patch4_window8_256.yaml │ │ ├── swin_mlp_tiny_c24_patch4_window8_256.yaml │ │ └── swin_mlp_tiny_c6_patch4_window8_256.yaml │ ├── swinmoe/ │ │ ├── swin_moe_base_patch4_window12_192_16expert_32gpu_22k.yaml │ │ ├── swin_moe_base_patch4_window12_192_32expert_32gpu_22k.yaml │ │ ├── swin_moe_base_patch4_window12_192_8expert_32gpu_22k.yaml │ │ ├── swin_moe_base_patch4_window12_192_cosine_router_32expert_32gpu_22k.yaml │ │ ├── swin_moe_base_patch4_window12_192_densebaseline_22k.yaml │ │ ├── swin_moe_small_patch4_window12_192_16expert_32gpu_22k.yaml │ │ ├── swin_moe_small_patch4_window12_192_32expert_32gpu_22k.yaml │ │ ├── swin_moe_small_patch4_window12_192_64expert_64gpu_22k.yaml │ │ ├── swin_moe_small_patch4_window12_192_8expert_32gpu_22k.yaml │ │ ├── swin_moe_small_patch4_window12_192_cosine_router_32expert_32gpu_22k.yaml │ │ └── swin_moe_small_patch4_window12_192_densebaseline_22k.yaml │ └── swinv2/ │ ├── swinv2_base_patch4_window12_192_22k.yaml │ ├── swinv2_base_patch4_window12to16_192to256_22kto1k_ft.yaml │ ├── swinv2_base_patch4_window12to24_192to384_22kto1k_ft.yaml │ ├── swinv2_base_patch4_window16_256.yaml │ ├── swinv2_base_patch4_window8_256.yaml │ ├── swinv2_large_patch4_window12_192_22k.yaml │ ├── swinv2_large_patch4_window12to16_192to256_22kto1k_ft.yaml │ ├── swinv2_large_patch4_window12to24_192to384_22kto1k_ft.yaml │ ├── swinv2_small_patch4_window16_256.yaml │ ├── swinv2_small_patch4_window8_256.yaml │ ├── swinv2_tiny_patch4_window16_256.yaml │ └── swinv2_tiny_patch4_window8_256.yaml ├── data/ │ ├── __init__.py │ ├── build.py │ ├── cached_image_folder.py │ ├── data_simmim_ft.py │ ├── data_simmim_pt.py │ ├── imagenet22k_dataset.py │ ├── map22kto1k.txt │ ├── samplers.py │ └── zipreader.py ├── get_started.md ├── kernels/ │ └── window_process/ │ ├── setup.py │ ├── swin_window_process.cpp │ ├── swin_window_process_kernel.cu │ ├── unit_test.py │ └── window_process.py ├── logger.py ├── lr_scheduler.py ├── main.py ├── main_moe.py ├── main_simmim_ft.py ├── main_simmim_pt.py ├── models/ │ ├── __init__.py │ ├── build.py │ ├── simmim.py │ ├── swin_mlp.py │ ├── swin_transformer.py │ ├── swin_transformer_moe.py │ └── swin_transformer_v2.py ├── optimizer.py ├── utils.py ├── utils_moe.py └── utils_simmim.py