diff --git a/models/yolov5s.yaml b/models/yolov5s.yaml
index aed2502..42678ac 100644
--- a/models/yolov5s.yaml
+++ b/models/yolov5s.yaml
@@ -49,26 +49,31 @@ backbone:
 #   ]
 
 # YOLOv5 v6.0 head (PAN + WeightedAdd)
-head: [
-  # ---------- Top-down FPN ----------
-  [-1, 1, Conv, [512, 1, 1]],                     # 10: reduce P5
-  [-1, 1, nn.Upsample, [None, 2, "nearest"]],     # 11
-  [[-1, 6], 1, WeightedAdd, [2]],                 # 12: P5 + backbone P4
-  [-1, 3, C3, [512, False]],                      # 13
+head:
+  [
+    # ---------- Top-down path ----------
+    [-1, 1, Conv, [512, 1, 1]],           # 10 reduce P5
+    [-1, 1, nn.Upsample, [None, 2, "nearest"]],  # 11
+    [6,  1, Conv, [512, 1, 1]],            # 12 align backbone P4
+    [[-1, -2], 1, WeightedAdd, [2]],       # 13 P5 + P4
+    [-1, 3, C3, [512, False]],             # 14
 
-  [-1, 1, Conv, [256, 1, 1]],                     # 14: reduce
-  [-1, 1, nn.Upsample, [None, 2, "nearest"]],     # 15
-  [[-1, 4], 1, WeightedAdd, [2]],                 # 16: P4 + backbone P3
-  [-1, 3, C3, [256, False]],                      # 17 (P3/8)
+    [-1, 1, Conv, [256, 1, 1]],            # 15 reduce
+    [-1, 1, nn.Upsample, [None, 2, "nearest"]],  # 16
+    [4,  1, Conv, [256, 1, 1]],            # 17 align backbone P3
+    [[-1, -2], 1, WeightedAdd, [2]],       # 18 P4 + P3
+    [-1, 3, C3, [256, False]],             # 19 (P3/8)
 
-  # ---------- Bottom-up PAN ----------
-  [-1, 1, Conv, [256, 3, 2]],                     # 18: downsample
-  [[-1, 13], 1, WeightedAdd, [2]],                # 19: P3 -> P4
-  [-1, 3, C3, [512, False]],                      # 20 (P4/16)
+    # ---------- Bottom-up path ----------
+    [-1, 1, Conv, [256, 3, 2]],            # 20 downsample P3 -> P4
+    [14, 1, Conv, [256, 1, 1]],             # 21 align top-down P4
+    [[-1, -2], 1, WeightedAdd, [2]],       # 22
+    [-1, 3, C3, [512, False]],             # 23 (P4/16)
 
-  [-1, 1, Conv, [512, 3, 2]],                     # 21: downsample
-  [[-1, 10], 1, WeightedAdd, [2]],                # 22: P4 -> P5
-  [-1, 3, C3, [1024, False]],                     # 23 (P5/32)
+    [-1, 1, Conv, [512, 3, 2]],            # 24 downsample P4 -> P5
+    [10, 1, Conv, [512, 1, 1]],             # 25 align P5
+    [[-1, -2], 1, WeightedAdd, [2]],       # 26
+    [-1, 3, C3, [1024, False]],            # 27 (P5/32)
 
-  [[17, 20, 23], 1, Detect, [nc, anchors]],       # Detect(P3, P4, P5)
-]
\ No newline at end of file
+    [[19, 23, 27], 1, Detect, [nc, anchors]],
+  ]
\ No newline at end of file
diff --git a/train_my_gra.sh b/train_my_gra.sh
index 5c036bb..e046438 100644
--- a/train_my_gra.sh
+++ b/train_my_gra.sh
@@ -2,9 +2,8 @@ export CUDA_VISIBLE_DEVICES=4,5,6,7
 python /data6/zhangw/yolov5/train.py \
     --img 960 \
     --batch 16 \
-    --epochs 300 \
+    --epochs 500 \
     --data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \
-    --weights /data6/zhangw/yolov5/yolov5s6.pt \
-    --project /data6/zhangw/ZW_gra_project/yolov5_result \
+    --project /data6/zhangw/ZW_gra_project/yolov5_result_structure \
     --cfg /data6/zhangw/yolov5/models/yolov5s.yaml \
-    --name G2_2_cub100_yolov5s6_changeNeck_e300
\ No newline at end of file
+    --name G2_2_cub100_yolov5s6_changeNeck_e500
\ No newline at end of file