mmpretrain/mmcls/apis
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
..
__init__.py [Enchance] Set a random seed when the user does not set a seed. (#554) 2021-12-02 18:09:55 +08:00
inference.py
test.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00
train.py [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +08:00