# 开始使用 --- 请事先参考[安装指南](install.md)配置运行环境 ## 1 设置环境变量 **设置PYTHONPATH环境变量:** ```bash export PYTHONPATH=path_to_PaddleClas:$PYTHONPATH ``` ## 2 模型训练与评估 PaddleClas 提供模型训练与评估脚本:tools/train.py和tools/eval.py ### 2.1 模型训练 ```bash # 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参数来更新配置 ```bash 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配置文件,具体配置参数参考[配置文档](config.md)。 ### 2.2 模型评估 ```bash 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参数更新配置。