mirrors/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-06-03 21:53:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

[doc]add amp train notes for detection train (#13481)

Browse Source 
* [doc]add amp train notes for detection train

* Update detection.md

* Update detection.md
This commit is contained in:
andyjiang 2024-07-24 14:51:05 +08:00 committed by GitHub
parent cf26f2330e
commit 9231cbe039
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
doc/doc_ch/detection.md
View File

@ -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>