getting_started.md 1.5 KB
Newer Older
W
WuHaobo 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
# 开始使用
---
请事先参考[安装指南](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 \
littletomatodonkey's avatar
littletomatodonkey 已提交
26
        -c ./configs/ResNet/ResNet50.yaml \
W
WuHaobo 已提交
27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
```

- 输出日志示例如下:

```
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 \
littletomatodonkey's avatar
littletomatodonkey 已提交
42
        -c ./configs/ResNet/ResNet50_vd.yaml \
W
WuHaobo 已提交
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
    -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 \
littletomatodonkey's avatar
littletomatodonkey 已提交
60
    -c ./configs/eval.yaml \
W
WuHaobo 已提交
61 62 63 64
    -o architecture="ResNet50_vd" \
    -o pretrained_model=path_to_pretrained_models
```
您可以更改configs/eval.yaml中的architecture字段和pretrained_model字段来配置评估模型,或是通过-o参数更新配置。