From 770a9723956064192e9ef9f254401c8f102f9c3e Mon Sep 17 00:00:00 2001
From: dongshuilong <dongshuilong@baidu.com>
Date: Wed, 23 Mar 2022 08:10:15 +0000
Subject: [PATCH] fix r50 benchmark static train yaml bug

---
 test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh
index 2b83ee8d..c4d38a98 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=""
-- 
GitLab