mmsegmentation/tools
Aston.He 7628a61f92
feat(mlu): Support PyTorch backend on MLU. (#1515)
* feat(mlu): Support PyTorch backend on MLU.

* fix redundant device variable.

* Update mmseg/apis/train.py

Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>

* Update comments.

* Update mmseg/apis/train.py

* Update is_mlu_available flag.

* align util_distribution.py to mmdet.

* align util_distribution.py to mmdet.

* add build_dp, build_ddp testcase.

* Update mmseg/utils/util_distribution.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update mmseg/utils/util_distribution.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update mmseg/utils/util_distribution.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update tests/test_utils/test_util_distribution.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update tests/test_utils/test_util_distribution.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update tests/test_utils/test_util_distribution.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update tests/test_utils/test_util_distribution.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* add mmcv version check for mlu device.

Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-25 18:11:42 +08:00
..
convert_datasets [fix] fix image save path bug in Windows (#1423) 2022-04-09 21:17:51 +08:00
model_converters [Feature] Add BEiT backbone (#1404) 2022-03-30 15:25:10 +08:00
torchserve [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
analyze_logs.py [Fix] Fix bug in `tools/analyse_logs.py` caused by wrong plot_iter in some cases. (#1428) 2022-04-05 12:47:41 +08:00
benchmark.py [Feature] benchmark can add work_dir and repeat times, test.py now has default work-dir (#1126) 2022-01-18 16:38:31 +08:00
browse_dataset.py [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
confusion_matrix.py [Fix] make arguments effective in tools/confusion_matrix.py (#1401) 2022-03-21 10:51:27 +08:00
deploy_test.py Add deprecation message for deploy tool (#1321) 2022-02-25 11:22:03 +08:00
dist_test.sh [Feature] Add multi machine `dist_train`. (#1383) 2022-03-18 10:55:49 +08:00
dist_train.sh [Enhancement] Delete seed in dist_train.sh (#1519) 2022-04-28 11:35:10 +08:00
get_flops.py Update get_flops.py (#1371) 2022-03-14 21:02:43 +08:00
onnx2tensorrt.py Add deprecation message for deploy tool (#1321) 2022-02-25 11:22:03 +08:00
print_config.py [Fix] revise --option to --options #1140 2021-12-15 15:24:08 +08:00
publish_model.py
pytorch2onnx.py Add deprecation message for deploy tool (#1321) 2022-02-25 11:22:03 +08:00
pytorch2torchscript.py [Enhancement] Add codespell pre-commit hook and fix typos (#920) 2021-10-13 06:21:17 -07:00
slurm_test.sh
slurm_train.sh
test.py feat(mlu): Support PyTorch backend on MLU. (#1515) 2022-05-25 18:11:42 +08:00
train.py feat(mlu): Support PyTorch backend on MLU. (#1515) 2022-05-25 18:11:42 +08:00