mmclassification/mmcls/models/utils
whcao affb39fe07
[Feature]Add Vit (#214)
* add imagenet bs 4096

* add vit_base_patch16_224_finetune

* add vit_base_patch16_224_pretrain

* add vit_base_patch16_384_finetune

* add vit_base_patch16_384_finetune

* add vit_b_p16_224_finetune_imagenet

* add vit_b_p16_224_pretrain_imagenet

* add vit_b_p16_384_finetune_imagenet

* add vit

* add vit

* add vit head

* vit unitest

* keep up with ClsHead

* test vit

* add flag to determiine whether to calculate acc during training

* Changes related to mmcv1.3.0

* change checkpoint saving interval to 10

* add label smooth

* default_runtime.py recovery

* docformatter

* docformatter

* delete 2 lines of comments

* delete configs/_base_/schedules/imagenet_bs4096.py

* add configs/_base_/schedules/imagenet_bs2048_AdamW.py

* rename imagenet_bs4096.py to imagenet_bs2048_AdamW.py

* add helpers.py

* test vit hybrid backbone

* fix HybridEmbed

* use to_2tuple instead
2021-04-16 19:22:41 +08:00
..
__init__.py [Feature]Add Vit (#214) 2021-04-16 19:22:41 +08:00
channel_shuffle.py Dev/backbone utils 2020-06-15 16:42:15 +08:00
cutmix.py [Fix] Fix version (#209) 2021-04-16 19:07:17 +08:00
helpers.py [Feature]Add Vit (#214) 2021-04-16 19:22:41 +08:00
inverted_residual.py Update CONTRIBUTING.md according to mmcv (#210) 2021-04-14 21:22:37 +08:00
make_divisible.py [Fix] Fix optional issues in docstring (#138) 2021-01-14 11:09:08 +08:00
mixup.py Update CONTRIBUTING.md according to mmcv (#210) 2021-04-14 21:22:37 +08:00
se_layer.py Dev mobilenetv3 2020-06-30 15:50:36 +08:00