Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

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

Merged
merged 3 commits into from
Jul 24, 2024

Conversation

andyjiang1116
Copy link
Collaborator

@andyjiang1116 andyjiang1116 commented Jul 24, 2024

add amp train notes for detection model, details see #12445

@andyjiang1116 andyjiang1116 requested a review from tink2123 July 24, 2024 05:49
@@ -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]中的解决方案进行使用。
Copy link
Collaborator

@tink2123 tink2123 Jul 24, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

文本检测模型使用AMP时可能遇到训练不收敛问题,可以参考#12445 中的临时解决方案进行使用。

Copy link
Collaborator

@tink2123 tink2123 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@GreatV GreatV merged commit 9231cbe into PaddlePaddle:main Jul 24, 2024
3 checks passed
# for free to join this conversation on GitHub. Already have an account? # to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants