diff --git a/docs/zh_CN/extension/train_on_xpu.md b/docs/zh_CN/extension/train_on_xpu.md new file mode 100644 index 000000000..2007686d4 --- /dev/null +++ b/docs/zh_CN/extension/train_on_xpu.md @@ -0,0 +1,20 @@ +# 图像分类昆仑模型介绍(持续更新中) + +## 前言 + +* 本文档介绍了目前昆仑支持的模型以及如何在昆仑设备上训练这些模型。支持昆仑的PaddlePaddle安装参考install_kunlun(https://github.com/PaddlePaddle/FluidDoc/blob/develop/doc/paddle/install/install_Kunlun_zh.md) + +## 昆仑训练 +* 数据来源和预训练模型参考[quick_start](../tutorials/quick_start.md)。昆仑训练效果与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```