From 5aea0790e34bbf34c2de5abb84ee138c05762095 Mon Sep 17 00:00:00 2001 From: andyjian1116 Date: Wed, 24 Jul 2024 13:07:53 +0800 Subject: [PATCH 1/3] [doc]add amp train notes for detection train --- doc/doc_ch/detection.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/doc/doc_ch/detection.md b/doc/doc_ch/detection.md index eba5213501a..814654e51c6 100644 --- a/doc/doc_ch/detection.md +++ b/doc/doc_ch/detection.md @@ -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]中的解决方案进行使用。 From 47adead085f5da6a9ccfb110ad5860a604618a13 Mon Sep 17 00:00:00 2001 From: andyjiang <87074272+andyjiang1116@users.noreply.github.com> Date: Wed, 24 Jul 2024 14:04:48 +0800 Subject: [PATCH 2/3] Update detection.md --- doc/doc_ch/detection.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc/doc_ch/detection.md b/doc/doc_ch/detection.md index 814654e51c6..f44a20e0738 100644 --- a/doc/doc_ch/detection.md +++ b/doc/doc_ch/detection.md @@ -142,7 +142,7 @@ python3 tools/train.py -c configs/det/det_mv3_db.yml \ Global.use_amp=True Global.scale_loss=1024.0 Global.use_dynamic_loss_scaling=True ``` **注意** -当前针对文本检测模型并未进行AMP精度验证，如果遇到训练不收敛问题，可以参考(discussions)[https://github.com/PaddlePaddle/PaddleOCR/discussions/12445]中的解决方案进行使用。 +文本检测模型使用AMP时可能遇到训练不收敛问题，可以参考(discussions)[https://github.com/PaddlePaddle/PaddleOCR/discussions/12445]中的临时解决方案进行使用。 From b2bb1643bd92d3f4b0bc563e2053fa71c55495b2 Mon Sep 17 00:00:00 2001 From: andyjiang <87074272+andyjiang1116@users.noreply.github.com> Date: Wed, 24 Jul 2024 14:27:19 +0800 Subject: [PATCH 3/3] Update detection.md --- doc/doc_ch/detection.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc/doc_ch/detection.md b/doc/doc_ch/detection.md index f44a20e0738..8271f100a93 100644 --- a/doc/doc_ch/detection.md +++ b/doc/doc_ch/detection.md @@ -142,7 +142,7 @@ python3 tools/train.py -c configs/det/det_mv3_db.yml \ Global.use_amp=True Global.scale_loss=1024.0 Global.use_dynamic_loss_scaling=True ``` **注意** -文本检测模型使用AMP时可能遇到训练不收敛问题，可以参考(discussions)[https://github.com/PaddlePaddle/PaddleOCR/discussions/12445]中的临时解决方案进行使用。 +文本检测模型使用AMP时可能遇到训练不收敛问题，可以参考[discussions](https://github.com/PaddlePaddle/PaddleOCR/discussions/12445)中的临时解决方案进行使用。