mmpretrain/tests/test_backbones
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
..
test_mobilenet_v2.py update shufflenet_v1 2020-07-12 00:06:55 +08:00
test_mobilenet_v3.py update shufflenet_v1 2020-07-12 00:06:55 +08:00
test_regnet.py Add RegNet 2020-07-01 14:21:45 +08:00
test_resnest.py Add ResNeSt (#25) 2020-09-22 10:41:51 +08:00
test_resnet.py Update CONTRIBUTING.md according to mmcv (#210) 2021-04-14 21:22:37 +08:00
test_resnet_cifar.py Add ResNet_CIFAR 2020-07-12 00:06:56 +08:00
test_resnext.py Refactoring for ResNet family 2020-06-25 11:57:50 +08:00
test_seresnet.py Update CONTRIBUTING.md according to mmcv (#210) 2021-04-14 21:22:37 +08:00
test_seresnext.py Fix the mid_channels of SEResNeXt 2020-07-07 11:25:48 +08:00
test_shufflenet_v1.py Add ResNeSt (#25) 2020-09-22 10:41:51 +08:00
test_shufflenet_v2.py update shufflenet_v1 2020-07-12 00:06:55 +08:00
test_utils.py [Feature] Add cutmix option (#198) 2021-04-14 21:27:42 +08:00
test_vgg.py Update CONTRIBUTING.md according to mmcv (#210) 2021-04-14 21:22:37 +08:00
test_vision_transformer.py [Feature]Add Vit (#214) 2021-04-16 19:22:41 +08:00