Ma Zerun
f9eb9b409b
[Docs] Add Copyright information. ( #413 )
2021-08-17 19:52:42 +08:00
Ma Zerun
1a7cebe4b9
[Refactor] Refactor unittest ( #321 )
...
* Refactor unit tests folder structure.
* Remove label smooth and Vit test in `test_classifiers.py`
* Rename test_utils in dataset to test_dataset_utils
* Split test_models/test_utils/test_utils.py to multiple sub files.
* Add unit tests of classifiers and heads
* Use patch context manager.
* Add unit test of `is_tracing`, and add warning in `is_tracing` if torch
verison is smaller than 1.6.0
2021-07-08 22:49:05 +08:00
Ma Zerun
71621a5f62
Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 ( #347 )
2021-07-07 11:55:53 +08:00
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