Ma Zerun
2b88df4484
[Refactor] Refactor the data flow. ( #989 )
...
* [Refactor] Refactor the data flow.
* Add comments about data preprocessor.
* Fix after mmengine folder structure refactoring.
2022-08-26 10:40:43 +08:00
mzr1996
24bcf069f8
[Refactor] Refactor dev scripts
2022-07-18 11:11:13 +08:00
Ezra-Yu
088d5b5add
[Refactor] Inherit all registries from MMEngine and use inherited registries.
2022-07-18 10:53:27 +08:00
Ma Zerun
c708770b42
[Enhance] Support dynamic input shape for ViT-based algorithms. ( #706 )
...
* Move `resize_pos_embed` to `mmcls.models.utils`
* Refactor Vision Transformer
* Refactor DeiT
* Refactor MLP-Mixer
* Refactor Swin-Transformer
* Remove `indexing` arg
* Support dynamic inputs for t2t_vit
* Add copyright
* Fix bugs in swin transformer
* Add `pad_small_maps` option
* Update swin transformer
* Handle `attn_mask` in checkpoints of swin
* Imporve by comments
2022-03-03 13:10:12 +08:00
Ma Zerun
f9a2b04cee
[Feature] Add DeiT backbone and checkpoints. ( #576 )
...
* Support DeiT backbone.
* Use hook to automatically resize pos embed
* Update ViT training setting
* Add deit configs and update docs
* Fix vit arch assertion
* Remove useless init function
* Add unit tests.
* Fix resize_pos_embed for DeiT
* Improve according to comments.
2021-12-15 22:44:57 +08:00