diff --git a/configs/Base-MGN.yml b/configs/Base-MGN.yml
index be56202..8c80832 100644
--- a/configs/Base-MGN.yml
+++ b/configs/Base-MGN.yml
@@ -9,9 +9,7 @@ MODEL:
     WITH_NL: False
 
   HEADS:
-    NAME: "ReductionHead"
-    POOL_LAYER: "maxpool"
-    REDUCTION_DIM: 256
+    EMBEDDING_DIM: 256
 
   LOSSES:
     NAME: ("CrossEntropyLoss", "TripletLoss",)
diff --git a/configs/Base-Strongerbaseline.yml b/configs/Base-Strongerbaseline.yml
index 81e6f03..36cc455 100644
--- a/configs/Base-Strongerbaseline.yml
+++ b/configs/Base-Strongerbaseline.yml
@@ -8,7 +8,7 @@ MODEL:
 
   HEADS:
     NECK_FEAT: "after"
-    POOL_LAYER: "gempool"
+    POOL_LAYER: "gempoolP"
     CLS_LAYER: "circleSoftmax"
     SCALE: 64
     MARGIN: 0.35
diff --git a/configs/Base-bagtricks.yml b/configs/Base-bagtricks.yml
index 1cc7fe1..43d2af2 100644
--- a/configs/Base-bagtricks.yml
+++ b/configs/Base-bagtricks.yml
@@ -6,12 +6,15 @@ MODEL:
     NORM: "BN"
     DEPTH: "50x"
     LAST_STRIDE: 1
+    FEAT_DIM: 2048
     WITH_IBN: False
     PRETRAIN: True
+    PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50-19c8e357.pth"
 
   HEADS:
-    NAME: "BNneckHead"
+    NAME: "EmbeddingHead"
     NORM: "BN"
+    WITH_BNNECK: True
     POOL_LAYER: "avgpool"
     NECK_FEAT: "before"
     CLS_LAYER: "linear"