PaddleClas/docs/zh_CN/training/advanced/train_on_xpu.md

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

---

目录

• 1. 前言
• 2. 昆仑训练
  • 2.1 ResNet50
  • 2.2 MobileNetV3
  • 2.3 HRNet
  • 2.4 VGG16/19

1. 前言

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

2. 昆仑训练

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

2.1 ResNet50

• 命令：

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

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

2.2 MobileNetV3

• 命令：

python3.7 ppcls/static/train.py \
    -c ppcls/configs/quick_start/MobileNetV3_large_x1_0.yaml \
    -o use_gpu=False \
    -o use_xpu=True \
    -o is_distributed=False

2.3 HRNet

• 命令：

python3.7 ppcls/static/train.py \
    -c ppcls/configs/quick_start/kunlun/HRNet_W18_C_finetune_kunlun.yaml \
    -o is_distributed=False \
    -o use_xpu=True \
    -o use_gpu=False

2.4 VGG16/19

• 命令：

python3.7 ppcls/static/train.py \
    -c ppcls/configs/quick_start/kunlun/VGG16_finetune_kunlun.yaml \
    -o use_gpu=False \
    -o use_xpu=True \
    -o is_distributed=False

python3.7 ppcls/static/train.py \
    -c ppcls/configs/quick_start/kunlun/VGG19_finetune_kunlun.yaml \
    -o use_gpu=False \
    -o use_xpu=True \
    -o is_distributed=False