Skip to content
New issue

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

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

Already on GitHub? Sign in 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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants