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

OSCHINA-MIRROR/monkeycc-mmsegmentation

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

NPU (华为 昇腾)

使用方法

请参考 MMCV 的安装文档 来安装 NPU 版本的 MMCV。

以下展示单机四卡场景的运行指令:

bash tools/dist_train.sh configs/deeplabv3/deeplabv3_r50-d8_4xb2-40k_cityscapes-512x1024.py 4

以下展示单机单卡下的运行指令:

python tools/train.py configs/deeplabv3/deeplabv3_r50-d8_4xb2-40k_cityscapes-512x1024.py

模型验证结果

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

注意:

  • 如果没有特别标记,NPU 上的使用混合精度训练的结果与使用 FP32 的 GPU 上的结果相同。

以上模型结果由华为昇腾团队提供

Опубликовать ( 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