✅ [Pass] the test of 2 new models

MultimediaTechLab · Jun 9, 2024 · 4e5e582 · 4e5e582
1 parent 459de68
commit 4e5e582
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yolo/config/model/v9-s.yaml b/yolo/config/model/v9-s.yaml
@@ -53,7 +53,7 @@ model:
             part_channels: 192
             csp_args: {repeat_num: 3}
         tags: N4
-
+  head:
     - UpSample:
         args: {scale_factor: 2, mode: nearest}
     - Concat:
@@ -96,7 +96,7 @@ model:
             reg_max: ${model.anchor.reg_max}
         output: True
 
-  head:
+  auxiliary:
     - SPPELAN:
         source: B5
         args: {out_channels: 256}