mmpretrain/mmcls/utils
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 [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +08:00
collect_env.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
device.py [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +08:00
distribution.py [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +08:00
logger.py [Docs] Refine API reference. (#774) 2022-04-15 20:19:20 +08:00
setup_env.py [Enhance] Add setup multi-processing both in train and test. (#671) 2022-01-27 10:18:36 +08:00