mmpretrain/tests/test_models
Ma Zerun 556fa567a8
[Feature] Support MViT and add checkpoints. (#924)
* [Feature] Support MViT.

* Add MViT configs and docs

* Add unit test

* Fix unit tests.
2022-08-02 15:20:16 +08:00
..
test_backbones [Feature] Support MViT and add checkpoints. (#924) 2022-08-02 15:20:16 +08:00
test_utils [Feature] Support resizemix. (#676) 2022-03-07 12:11:20 +08:00
test_classifiers.py [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
test_heads.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
test_neck.py [Feature] Support gem pooling (#677) 2022-02-17 02:17:36 +08:00