mmclassification/tests/test_models/test_utils
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
..
test_attention.py [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00
test_augment.py [Enhance] Suport Mixup&Cutmix for multi-label task. 2022-01-21 11:30:58 +08:00
test_embed.py
test_inverted_residual.py
test_misc.py
test_position_encoding.py [Feature] Add Twins backbone and convert checkpoints. (#642) 2022-01-27 10:25:05 +08:00
test_se.py [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00