1.1 KiB

Raw Blame History

图像分类昆仑模型介绍(持续更新中)

前言

本文档介绍了目前昆仑支持的模型以及如何在昆仑设备上训练这些模型。支持昆仑的PaddlePaddle安装参考install_kunlun(https://github.com/PaddlePaddle/FluidDoc/blob/develop/doc/paddle/install/install_Kunlun_zh.md)

昆仑训练

数据来源和预训练模型参考quick_start。昆仑训练效果与CPU/GPU对齐。

ResNet50

命令：

python3.7 tools/static/train.py -c configs/quick_start/ResNet50_vd_finetune_kunlun.yaml -o use_gpu=False -o use_xpu=True -o is_distributed=False

与cpu/gpu训练的区别是加上-o use_xpu=True, 表示执行在昆仑设备上。

MobileNetV3

命令：

python3.7 tools/static/train.py -c configs/quick_start/MobileNetV3_large_x1_0_finetune.yaml -o use_gpu=False -o use_xpu=True -o is_distributed=False

HRNet

命令：

python3.7 tools/static/train.py -c configs/quick_start/HRNet_W18_C_finetune.yaml -o is_distributed=False -o use_cpu=False -o use_xpu=True -o use_gpu=False

1.1 KiB Raw Blame History