diff --git a/test_tipc/configs/BMN/train_dy2static_python.txt b/test_tipc/configs/BMN/train_dy2static_python.txt
index 2796c954e..6c5804262 100644
--- a/test_tipc/configs/BMN/train_dy2static_python.txt
+++ b/test_tipc/configs/BMN/train_dy2static_python.txt
@@ -13,7 +13,7 @@ train_model_name:null
 -o DATASET.train.file_path:null
 ##
 trainer:norm_train
-norm_train:main.py --validate -c configs/localization/bmn.yaml --seed 1234
+norm_train:main.py -c configs/localization/bmn.yaml --seed 1234 --max_iters=3 -o log_interval=1
 pact_train:null
 fpgm_train:null
 distill_train:null
diff --git a/test_tipc/test_train_dy2static_python.sh b/test_tipc/test_train_dy2static_python.sh
index cdb9ebdd3..b82f3eb04 100644
--- a/test_tipc/test_train_dy2static_python.sh
+++ b/test_tipc/test_train_dy2static_python.sh
@@ -54,8 +54,8 @@ do
     eval $cmd
 
     # analysis and compare the losses.
-    dyout=`cat $dy2static_output | python test_tipc/extract_loss.py -v 'train step' -e 'loss: {%f} '`
-    stout=`cat $dygraph_output   | python test_tipc/extract_loss.py -v 'train step' -e 'loss: {%f} '`
+    dyout=`cat $dy2static_output | python test_tipc/extract_loss.py -v 'train step' -e 'loss: {%f} ' | head -n 3`
+    stout=`cat $dygraph_output   | python test_tipc/extract_loss.py -v 'train step' -e 'loss: {%f} ' | head -n 3`
     echo $dyout > $dygraph_loss
     echo $stout > $dy2static_loss
     diff_log=$LOG_PATH/${config_name}_diff_log.txt