Skip to content

GitLab

Commit 4287fd21 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

add model_name to save_log

parent c1cba283
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test_tipc/test_train_inference_python.sh
View file @ 4287fd21
...@@ -316,14 +316,14 @@ else ...@@ -316,14 +316,14 @@ else
set_train_params1=$(func_set_params "${train_param_key1}" "${train_param_value1}") set_train_params1=$(func_set_params "${train_param_key1}" "${train_param_value1}")
set_use_gpu=$(func_set_params "${train_use_gpu_key}" "${train_use_gpu}") set_use_gpu=$(func_set_params "${train_use_gpu_key}" "${train_use_gpu}")
if [ ${#ips} -le 26 ];then if [ ${#ips} -le 26 ];then
save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}" save_log="${LOG_PATH}/${model_name}/${trainer}_gpus_${gpu}_autocast_${autocast}"
nodes=1 nodes=1
else else
IFS="," IFS=","
ips_array=(${ips}) ips_array=(${ips})
IFS="|" IFS="|"
nodes=${#ips_array[@]} nodes=${#ips_array[@]}
save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}_nodes_${nodes}" save_log="${LOG_PATH}/${model_name}/${trainer}_gpus_${gpu}_autocast_${autocast}_nodes_${nodes}"
fi fi
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment