mmengine/mmengine
黄启元 60b4c199fc
[Feature] Support MLU backend (#1075)
* support mlu device

* support mlu device

* fix lint error

* fix lint error builder.py

* fix lint error in amp.py

* fix lint errors

* fix data type in instance_data.py
2023-04-14 19:06:19 +08:00
..
analysis [Enhance] Complement type hint of get_model_complexity_info() (#1064) 2023-04-10 19:57:54 +08:00
config [Enhance] Add `mmpretrain` to the `MODULE2PACKAGE` (#1002) 2023-03-17 11:03:44 +08:00
dataset [Fix] Update the ClassBalancedDataset logic to keep len(repeat_factors) = len(dataset) (#1048) 2023-04-04 14:27:27 +08:00
device [Refactor] Refactor the `get_device` and `is_npu_available` (#1004) 2023-03-19 15:55:27 +08:00
dist [Feature] Support MLU backend (#1075) 2023-04-14 19:06:19 +08:00
evaluator [Enhancement] Replace warnings.warn with print_log (#961) 2023-03-06 17:25:28 +08:00
fileio [Enhancement] Replace warnings.warn with print_log (#961) 2023-03-06 17:25:28 +08:00
hooks [Fix] Fix publish multiple checkpoints when using multiple GPUs (#1059) (#1070) 2023-04-12 10:38:48 +08:00
hub
infer [Fix] Fix inferencer gets wrong configs path (#996) 2023-03-14 18:28:33 +08:00
logging [Enhancement] Support dumping logs of different ranks (#968) 2023-03-13 14:35:31 +08:00
model [Feature] Support MLU backend (#1075) 2023-04-14 19:06:19 +08:00
optim [Feature] Support MLU backend (#1075) 2023-04-14 19:06:19 +08:00
registry [Feature] Support registering partial functions and more (#595) 2023-04-10 19:42:04 +08:00
runner [Feature] Support MLU backend (#1075) 2023-04-14 19:06:19 +08:00
structures [Feature] Support MLU backend (#1075) 2023-04-14 19:06:19 +08:00
testing
utils [Enhancement] Silence error when `ManagerMixin` built instance with duplicate name. (#990) 2023-03-10 11:07:31 +08:00
visualization [Feature] Add vis backend for MLflow. (#878) 2023-04-07 16:35:41 +08:00
__init__.py
version.py Bump version to v0.7.2 (#1051) 2023-04-06 17:12:00 +08:00