File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments