1 В избранное 0 Ответвления 0

OSCHINA-MIRROR/monkeycc-mmsegmentation

Присоединиться к Gitlife
Откройте для себя и примите участие в публичных проектах с открытым исходным кодом с участием более 10 миллионов разработчиков. Приватные репозитории также полностью бесплатны :)
Присоединиться бесплатно
Клонировать/Скачать
npu.md 4.8 КБ
Копировать Редактировать Web IDE Исходные данные Просмотреть построчно История
wangjiangben-hw Отправлено 16.02.2023 12:43 e9d5b03

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

Model mIoU Config Download
deeplabv3 78.85 config log
deeplabv3plus 79.23 config log
hrnet 78.1 config log
fcn 74.15 config log
icnet 69.25 config log
pspnet 77.21 config log
unet 68.86 config log
upernet 77.81 config log
apcnet 78.02 config log
bisenetv1 76.04 config log
bisenetv2 72.44 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.

Опубликовать ( 0 )

Вы можете оставить комментарий после Вход в систему

1
https://api.gitlife.ru/oschina-mirror/monkeycc-mmsegmentation.git
git@api.gitlife.ru:oschina-mirror/monkeycc-mmsegmentation.git
oschina-mirror
monkeycc-mmsegmentation
monkeycc-mmsegmentation
main