diff --git a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh
index 2b83ee8d69d64d796bf3d19abf8453f128fc21f4..c4d38a98e9101cec69645700bf3fd56ddc11d63a 100644
--- a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh
+++ b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh
@@ -35,9 +35,9 @@ function _train(){
     if [ ${fp_item} = "fp32" ]; then
         config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50.yaml"
     elif [ ${fp_item} = "amp_fp16" ]; then
-        config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1.yaml"
+        config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O1_ultra.yaml"
     elif [ ${fp_item} = "pure_fp16" ]; then
-    	config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2.yaml"
+    	config_file="-c ppcls/configs/ImageNet/ResNet/ResNet50_amp_O2_ultra.yaml"
     fi
     if [ ${profiling} = "false" ]; then
 	profiling_config=""