Skip to content

Commit

Permalink
[doc]add amp train notes for detection train (#13481)
Browse files Browse the repository at this point in the history
* [doc]add amp train notes for detection train

* Update detection.md

* Update detection.md
  • Loading branch information
andyjiang1116 authored Jul 24, 2024
1 parent cf26f23 commit 9231cbe
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions doc/doc_ch/detection.md
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,8 @@ python3 tools/train.py -c configs/det/det_mv3_db.yml \
-o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained \
Global.use_amp=True Global.scale_loss=1024.0 Global.use_dynamic_loss_scaling=True
```
**注意**
文本检测模型使用AMP时可能遇到训练不收敛问题,可以参考[discussions](https://github.com/PaddlePaddle/PaddleOCR/discussions/12445)中的临时解决方案进行使用。

<a name="25---fleet---"></a>

Expand Down

0 comments on commit 9231cbe

Please sign in to comment.