mmclassification/mmcls/models/heads
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
base_head.py Update CONTRIBUTING.md according to mmcv (#210) 2021-04-14 21:22:37 +08:00
cls_head.py [Feature]Add cal_acc to cls_head.py (#206) 2021-04-13 13:52:14 +08:00
linear_head.py
multi_label_head.py Update CONTRIBUTING.md according to mmcv (#210) 2021-04-14 21:22:37 +08:00
multi_label_linear_head.py Update CONTRIBUTING.md according to mmcv (#210) 2021-04-14 21:22:37 +08:00
vision_transformer_head.py [Feature]Add Vit (#214) 2021-04-16 19:22:41 +08:00