Skip to content

Commit d436ab7

Browse files
authored
[Benchmark] Quick fix for benchmark script (PaddlePaddle#1268)
1 parent a6094c4 commit d436ab7

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/benchmark/run_benchmark.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,6 @@ function _set_params(){
1414
need_profile=${7:-"off"}
1515
gpt_repo=${8:-"gpt2"}
1616

17-
base_batch_size=$(($batch_size*1024))
18-
model_name=${model_name}_${gpt_repo}_${run_mode}_bs${batch_size}_${fp_item}_${num_gpu_devices}_${dygraph_name}
1917
mission_name="语义表示"
2018
direction_id=1
2119
run_log_path=${TRAIN_LOG_DIR:-$(pwd)}
@@ -24,7 +22,8 @@ function _set_params(){
2422
arr=(${device})
2523
num_gpu_devices=${#arr[*]}
2624

27-
25+
base_batch_size=$(($batch_size*1024))
26+
model_name=${model_name}_${gpt_repo}_${run_mode}_bs${batch_size}_${fp_item}_${num_gpu_devices}_${dygraph_name}
2827
log_file=${run_log_path}/${model_name}
2928
log_folder=${run_log_path}/${model_name}_logdir
3029
log_profile=${run_log_path}/${model_name}_model.profile

0 commit comments

Comments
 (0)