未验证 提交 9b626dda 编写于 作者: S shangliang Xu 提交者: GitHub

fix ppyoloe amp bug, add reduce_mean to custom_black_list (#6787)

上级 74373cc6
...@@ -78,12 +78,12 @@ The PaddleDetection team provides configs and weights of various feature detecti ...@@ -78,12 +78,12 @@ The PaddleDetection team provides configs and weights of various feature detecti
Training PP-YOLOE+ on 8 GPUs with following command Training PP-YOLOE+ on 8 GPUs with following command
```bash ```bash
python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml --eval --amp
``` ```
**Notes:** **Notes:**
- use `--amp` to train with default config to avoid out of memeory.
- If you need to evaluate while training, please add `--eval`. - If you need to evaluate while training, please add `--eval`.
- PP-YOLOE+ supports mixed precision training, please add `--amp`.
- PaddleDetection supports multi-machine distribued training, you can refer to [DistributedTraining tutorial](../../docs/DistributedTraining_en.md). - PaddleDetection supports multi-machine distribued training, you can refer to [DistributedTraining tutorial](../../docs/DistributedTraining_en.md).
......
...@@ -78,10 +78,11 @@ PaddleDetection团队提供了基于PP-YOLOE的各种垂类检测模型的配置 ...@@ -78,10 +78,11 @@ PaddleDetection团队提供了基于PP-YOLOE的各种垂类检测模型的配置
请执行以下指令训练PP-YOLOE+ 请执行以下指令训练PP-YOLOE+
```bash ```bash
python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml --eval --amp
``` ```
**注意:** **注意:**
- 如果需要边训练边评估,请添加`--eval`. - 如果需要边训练边评估,请添加`--eval`.
- PP-YOLOE+支持混合精度训练,请添加`--amp`.
- PaddleDetection支持多机训练,可以参考[多机训练教程](../../docs/DistributedTraining_cn.md). - PaddleDetection支持多机训练,可以参考[多机训练教程](../../docs/DistributedTraining_cn.md).
### 评估 ### 评估
......
...@@ -2,6 +2,7 @@ architecture: YOLOv3 ...@@ -2,6 +2,7 @@ architecture: YOLOv3
norm_type: sync_bn norm_type: sync_bn
use_ema: true use_ema: true
ema_decay: 0.9998 ema_decay: 0.9998
custom_black_list: ['reduce_mean']
YOLOv3: YOLOv3:
backbone: CSPResNet backbone: CSPResNet
......
...@@ -2,6 +2,7 @@ architecture: YOLOv3 ...@@ -2,6 +2,7 @@ architecture: YOLOv3
norm_type: sync_bn norm_type: sync_bn
use_ema: true use_ema: true
ema_decay: 0.9998 ema_decay: 0.9998
custom_black_list: ['reduce_mean']
YOLOv3: YOLOv3:
backbone: CSPResNet backbone: CSPResNet
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册