2020-12-16 20:58:26 +08:00
|
|
|
# 图像分类昆仑模型介绍(持续更新中)
|
|
|
|
|
|
|
|
|
|
## 前言
|
|
|
|
|
|
|
|
|
|
* 本文档介绍了目前昆仑支持的模型以及如何在昆仑设备上训练这些模型。支持昆仑的PaddlePaddle安装参考install_kunlun(https://github.com/PaddlePaddle/FluidDoc/blob/develop/doc/paddle/install/install_Kunlun_zh.md)
|
|
|
|
|
|
|
|
|
|
## 昆仑训练
|
2021-04-23 13:00:46 +08:00
|
|
|
* 数据来源和预训练模型参考[quick_start](../tutorials/quick_start_new_user.md)。昆仑训练效果与CPU/GPU对齐。
|
2020-12-16 20:58:26 +08:00
|
|
|
|
|
|
|
|
### 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```
|
2020-12-29 17:36:32 +08:00
|
|
|
|
|
|
|
|
### 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```
|
2021-01-15 16:24:12 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
### 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```
|
