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

leo-q8 · web-flow · commit f4399e8d3f94 · 2024-11-15T13:41:05.000+08:00
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
