提交 144aa589 编写于 作者: G gaotingquan

docs: update docs of training on xpu

上级 8ac51f74
......@@ -5,28 +5,28 @@
* 本文档介绍了目前昆仑支持的模型以及如何在昆仑设备上训练这些模型。支持昆仑的PaddlePaddle安装参考install_kunlun(https://github.com/PaddlePaddle/FluidDoc/blob/develop/doc/paddle/install/install_Kunlun_zh.md)
## 昆仑训练
* 数据来源和预训练模型参考[quick_start](../tutorials/quick_start_new_user.md)。昆仑训练效果与CPU/GPU对齐。
* 数据来源和预训练模型参考[quick_start](../quick_start/quick_start_classification_new_user.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```
```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, 表示执行在昆仑设备上。
### 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```
```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```
### 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```
```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```
### VGG16/19
* 命令:
```python3.7 tools/static/train.py -c configs/quick_start/VGG16_finetune_kunlun.yaml -o use_gpu=False -o use_cpu=False -o use_xpu=True -o is_distributed=False```
```python3.7 tools/static/train.py -c configs/quick_start/VGG19_finetune_kunlun.yaml -o use_gpu=False -o use_cpu=False -o use_xpu=True -o is_distributed=False```
```python3.7 ppcls/static/train.py -c ppcls/configs/quick_start/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/VGG19_finetune_kunlun.yaml -o use_gpu=False -o use_xpu=True -o is_distributed=False```
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册