PaddleClas/docs/zh_cn/tutorials/getting_started.md

开始使用

---

请事先参考安装指南配置运行环境

1 设置环境变量

设置PYTHONPATH环境变量：

export PYTHONPATH=path_to_PaddleClas:$PYTHONPATH

2 模型训练与评估

PaddleClas 提供模型训练与评估脚本：tools/train.py和tools/eval.py

2.1 模型训练

# PaddleClas通过launch方式启动多卡多进程训练
# 通过设置FLAGS_selected_gpus 指定GPU运行卡号

python -m paddle.distributed.launch \
    --selected_gpus="0,1,2,3" \
    --log_dir=log_ResNet50 \
    train.py \
        -c ../configs/ResNet/ResNet50.yaml \

• 输出日志示例如下：

epoch:0    train    step:13    loss:7.9561    top1:0.0156    top5:0.1094    lr:0.100000    elapse:0.193

可以通过添加-o参数来更新配置

python -m paddle.distributed.launch \
    --selected_gpus="0,1,2,3" \
    --log_dir=log_ResNet50_vd \
    train.py \
        -c ../configs/ResNet/ResNet50_vd.yaml \
    -o use_mix=1 \

• 输出日志示例如下：

epoch:0    train    step:522    loss:1.6330    lr:0.100000    elapse:0.210

或是直接修改模型对应的yaml配置文件，具体配置参数参考配置文档。

2.2 模型评估

python eval.py \
    -c ../configs/eval.yaml \
    -o architecture="ResNet50_vd" \
    -o pretrained_model=path_to_pretrained_models

您可以更改configs/eval.yaml中的architecture字段和pretrained_model字段来配置评估模型，或是通过-o参数更新配置。