mmclassification/tests/test_models
Ma Zerun c03efeeea4
[Feature] Support MPS device. (#894)
* [Feature] Support MPS device.

* Add `auto_select_device`

* Add unit tests
2022-07-28 12:28:51 +08:00
..
test_backbones [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +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