File tree
2 files changed
+2
-2
lines changed- 3rdparty
- examples/pytorch/swin/Swin-Transformer-Quantization
2 files changed
+2
-2
lines changedSubmodule Megatron-LM updated 106 files
Submodule SwinTransformer updated 62 files
- .gitignore+6
- MODELHUB.md+71
- README.md+88-24
- config.py+64-4
- configs/swin/swin_base_patch4_window12_384_22kto1k_finetune.yaml
- configs/swin/swin_base_patch4_window12_384_finetune.yaml
- configs/swin/swin_base_patch4_window7_224.yaml
- configs/swin/swin_base_patch4_window7_224_22k.yaml
- configs/swin/swin_base_patch4_window7_224_22kto1k_finetune.yaml
- configs/swin/swin_large_patch4_window12_384_22kto1k_finetune.yaml
- configs/swin/swin_large_patch4_window7_224_22k.yaml
- configs/swin/swin_large_patch4_window7_224_22kto1k_finetune.yaml
- configs/swin/swin_small_patch4_window7_224.yaml
- configs/swin/swin_small_patch4_window7_224_22k.yaml+18
- configs/swin/swin_small_patch4_window7_224_22kto1k_finetune.yaml+16
- configs/swin/swin_tiny_c24_patch4_window8_256.yaml
- configs/swin/swin_tiny_patch4_window7_224.yaml
- configs/swin/swin_tiny_patch4_window7_224_22k.yaml+18
- configs/swin/swin_tiny_patch4_window7_224_22kto1k_finetune.yaml+16
- configs/swinmlp/swin_mlp_base_patch4_window7_224.yaml
- configs/swinmlp/swin_mlp_tiny_c12_patch4_window8_256.yaml
- configs/swinmlp/swin_mlp_tiny_c24_patch4_window8_256.yaml
- configs/swinmlp/swin_mlp_tiny_c6_patch4_window8_256.yaml
- configs/swinmoe/swin_moe_base_patch4_window12_192_16expert_32gpu_22k.yaml+31
- configs/swinmoe/swin_moe_base_patch4_window12_192_32expert_32gpu_22k.yaml+31
- configs/swinmoe/swin_moe_base_patch4_window12_192_8expert_32gpu_22k.yaml+31
- configs/swinmoe/swin_moe_base_patch4_window12_192_cosine_router_32expert_32gpu_22k.yaml+32
- configs/swinmoe/swin_moe_base_patch4_window12_192_densebaseline_22k.yaml+26
- configs/swinmoe/swin_moe_small_patch4_window12_192_16expert_32gpu_22k.yaml+31
- configs/swinmoe/swin_moe_small_patch4_window12_192_32expert_32gpu_22k.yaml+31
- configs/swinmoe/swin_moe_small_patch4_window12_192_64expert_64gpu_22k.yaml+31
- configs/swinmoe/swin_moe_small_patch4_window12_192_8expert_32gpu_22k.yaml+31
- configs/swinmoe/swin_moe_small_patch4_window12_192_cosine_router_32expert_32gpu_22k.yaml+32
- configs/swinmoe/swin_moe_small_patch4_window12_192_densebaseline_22k.yaml+26
- configs/swinv2/swinv2_base_patch4_window12_192_22k.yaml+19
- configs/swinv2/swinv2_base_patch4_window12to16_192to256_22kto1k_ft.yaml+19
- configs/swinv2/swinv2_base_patch4_window12to24_192to384_22kto1k_ft.yaml+21
- configs/swinv2/swinv2_base_patch4_window16_256.yaml+11
- configs/swinv2/swinv2_base_patch4_window8_256.yaml+11
- configs/swinv2/swinv2_large_patch4_window12_192_22k.yaml+19
- configs/swinv2/swinv2_large_patch4_window12to16_192to256_22kto1k_ft.yaml+19
- configs/swinv2/swinv2_large_patch4_window12to24_192to384_22kto1k_ft.yaml+21
- configs/swinv2/swinv2_small_patch4_window16_256.yaml+11
- configs/swinv2/swinv2_small_patch4_window8_256.yaml+11
- configs/swinv2/swinv2_tiny_patch4_window16_256.yaml+11
- configs/swinv2/swinv2_tiny_patch4_window8_256.yaml+11
- data/build.py+14-2
- data/imagenet22k_dataset.py+55
- get_started.md+84-39
- kernels/window_process/setup.py+12
- kernels/window_process/swin_window_process.cpp+132
- kernels/window_process/swin_window_process_kernel.cu+323
- kernels/window_process/unit_test.py+250
- kernels/window_process/window_process.py+63
- main.py+54-68
- main_moe.py+367
- models/build.py+54-1
- models/swin_transformer.py+40-11
- models/swin_transformer_moe.py+824
- models/swin_transformer_v2.py+633
- utils.py+51-12
- utils_moe.py+241
0 commit comments