mmpretrain/mmcls/utils
TangYueran dc8691e889
[Feature] Support MLU backend. (#1159)
* Training on MLU is available
2022-11-15 17:02:16 +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 MLU backend. (#1159) 2022-11-15 17:02:16 +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