mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00

[Fix]Fix config issue in readme. (#3047 )

Thanks for your contribution and we appreciate it a lot. The following
instructions would make your pull request more healthy and more easily
get feedback. If you do not understand some items, don't worry, just
make the pull request and seek help from maintainers.

## Motivation

Fix uprenet config issue in readme.

## Modification

Fix npu.md.

## BC-breaking (Optional)

None.

## Use cases (Optional)

None.

## Checklist

1. Pre-commit or other linting tools are used to fix the potential lint
issues.
2. The modification is covered by complete unit tests. If not, please
add more unit test to ensure the correctness.
3. If the modification has potential influence on downstream projects,
this PR should be tested with downstream projects, like MMDet or
MMDet3D.
4. The documentation has been modified accordingly, like docstring or
example tutorials.

2023-05-26 14:37:44 +08:00

3.8 KiB

Raw Blame History

NPU (HUAWEI Ascend)

Usage

Please refer to the building documentation of MMCV to install MMCV and MMEngine on NPU devices.

Here we use 4 NPUs on your computer to train the model with the following command:

bash tools/dist_train.sh configs/deeplabv3/deeplabv3_r50-d8_512x1024_40k_cityscapes.py 4

Also, you can use only one NPU to train the model with the following command:

python tools/train.py configs/deeplabv3/deeplabv3_r50-d8_512x1024_40k_cityscapes.py

Models Results

Model	mIoU	Config	Download
deeplabv3	78.92	config	log
deeplabv3plus	79.68	config	log
hrnet	77.09	config	log
fcn	72.69	config	log
pspnet	78.07	config	log
unet	69.00	config	log
apcnet	78.07	config	log
upernet	78.15	config	log

Notes:

If not specially marked, the results on NPU with amp are the basically same as those on the GPU with FP32.

All above models are provided by Huawei Ascend group.

3.8 KiB Raw Blame History

NPU (HUAWEI Ascend)

Usage

Models Results

3.8 KiB

Raw Blame History