fix pp-yoloe-sod models train bug (#2487)

PaddlePaddle · Nov 15, 2024 · f4399e8 · f4399e8
1 parent 752a814
commit f4399e8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-L.yaml b/paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-L.yaml
@@ -86,6 +86,7 @@ use_ema: true
 ema_decay: 0.9998
 ema_black_list: ['proj_conv.weight']
 custom_black_list: ['reduce_mean']
+find_unused_parameters: True
 
 YOLOv3:
   backbone: CSPResNet

diff --git a/paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-S.yaml b/paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-S.yaml
@@ -86,6 +86,7 @@ use_ema: true
 ema_decay: 0.9998
 ema_black_list: ['proj_conv.weight']
 custom_black_list: ['reduce_mean']
+find_unused_parameters: True
 
 YOLOv3:
   backbone: CSPResNet

diff --git a/paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-largesize-L.yaml b/paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-largesize-L.yaml
@@ -88,6 +88,7 @@ use_ema: true
 ema_decay: 0.9998
 ema_black_list: ['proj_conv.weight']
 custom_black_list: ['reduce_mean']
+find_unused_parameters: True
 
 YOLOv3:
   backbone: CSPResNet