mmpretrain/tests
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
..
data [CI] Add Windows CI. (#708) 2022-02-28 10:06:08 +08:00
test_data [Feature] Support MViT and add checkpoints. (#924) 2022-08-02 15:20:16 +08:00
test_downstream [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00
test_metrics [Enhance] Use PyTorch official `one_hot` to implement `convert_to_one_hot`. (#696) 2022-02-28 10:20:31 +08:00
test_models [Feature] Support MViT and add checkpoints. (#924) 2022-08-02 15:20:16 +08:00
test_runtime [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00
test_utils [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +08:00