Prechádzať zdrojové kódy

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

学卿 1 rok pred
rodič
commit
f4399e8d3f

+ 1 - 0
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

+ 1 - 0
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

+ 1 - 0
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