NPU (HUAWEI Ascend)

Usage

Please refer to the building documentation of MMCV to install MMCV 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_4xb2-40k_cityscapes-512x1024.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_4xb2-40k_cityscapes-512x1024.py

Models Results

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.