mmpretrain/tests
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
..
data
test_backbones [Feature]Add Vit (#214) 2021-04-16 19:22:41 +08:00
test_pipelines [Feature] Support random augmentation (#201) 2021-04-09 14:02:50 +08:00
test_classifiers.py [Feature]Add Vit (#214) 2021-04-16 19:22:41 +08:00
test_dataset.py [Feature] Add heads and config for multilabel task (#145) 2021-01-25 18:10:14 +08:00
test_eval_hook.py [Fix] Use MMCV's EvalHook in MMClassification (#182) 2021-03-25 17:38:51 +08:00
test_heads.py [Feature]Add cal_acc to cls_head.py (#206) 2021-04-13 13:52:14 +08:00
test_losses.py [Bug]Fix label smooth bug (#203) 2021-04-13 13:53:56 +08:00
test_metrics.py Add evaluation metrics for multilabel task (#123) 2021-01-04 12:25:33 +08:00