diff --git a/projects/FastRetri/configs/base-image_retri.yml b/projects/FastRetri/configs/base-image_retri.yml
index cdf2bc9..66e4353 100644
--- a/projects/FastRetri/configs/base-image_retri.yml
+++ b/projects/FastRetri/configs/base-image_retri.yml
@@ -12,6 +12,7 @@ MODEL:
   HEADS:
     NAME: EmbeddingHead
     NORM: syncBN
+    WITH_BNNECK: True
     NECK_FEAT: after
     EMBEDDING_DIM: 0
     POOL_LAYER: GeneralizedMeanPooling
@@ -52,7 +53,7 @@ DATALOADER:
 SOLVER:
   MAX_EPOCH: 100
   AMP:
-    ENABLED: False
+    ENABLED: True
 
   OPT: SGD
   SCHED: CosineAnnealingLR
@@ -73,6 +74,9 @@ SOLVER:
 
   CHECKPOINT_PERIOD: 10
 
+  CLIP_GRADIENTS:
+    ENABLED: True
+
 TEST:
   EVAL_PERIOD: 10
   IMS_PER_BATCH: 256
diff --git a/projects/FastRetri/configs/cub.yml b/projects/FastRetri/configs/cub.yml
index 2d0cdf7..fce0ba3 100644
--- a/projects/FastRetri/configs/cub.yml
+++ b/projects/FastRetri/configs/cub.yml
@@ -3,7 +3,7 @@ _BASE_: base-image_retri.yml
 MODEL:
   LOSSES:
     CE:
-      EPSILON: 0.4
+      EPSILON: 0.3
 
 INPUT:
   SIZE_TRAIN: [256,]