diff --git a/benchmarks/test_tipc/ernie/dygraph/hybrid_parallel/N1C1/ernie_bs16_fp32_DP1-MP1-PP1.sh b/benchmarks/test_tipc/ernie/dygraph/hybrid_parallel/N1C1/ernie_bs16_fp32_DP1-MP1-PP1.sh index 521a09ca0..439375ee4 100644 --- a/benchmarks/test_tipc/ernie/dygraph/hybrid_parallel/N1C1/ernie_bs16_fp32_DP1-MP1-PP1.sh +++ b/benchmarks/test_tipc/ernie/dygraph/hybrid_parallel/N1C1/ernie_bs16_fp32_DP1-MP1-PP1.sh @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -model_item=gpt +model_item=ernie dp_degree=1 mp_degree=1 pp_degree=1 @@ -21,7 +21,7 @@ fp_item=fp32 run_mode=DP1-MP1-PP1 device_num=N1C1 -model=gpt +model=ernie micro_bs=${bs_item} cd ./benchmarks