From 9d0e0e20c42f784cc77440d1a27ab75214fe4c88 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Fri, 4 Mar 2022 11:23:41 +0800 Subject: [PATCH] add model_name --- test_tipc/test_train_inference_python.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh index 16024caf..fe98cb00 100644 --- a/test_tipc/test_train_inference_python.sh +++ b/test_tipc/test_train_inference_python.sh @@ -125,7 +125,7 @@ if [ ${MODE} = "klquant_whole_infer" ]; then infer_value1=$(func_parser_value "${lines[19]}") fi -LOG_PATH="./test_tipc/output" +LOG_PATH="./test_tipc/output/${model_name}" mkdir -p ${LOG_PATH} status_log="${LOG_PATH}/results_python.log" @@ -316,14 +316,14 @@ else 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}") if [ ${#ips} -le 26 ];then - save_log="${LOG_PATH}/${model_name}/${trainer}_gpus_${gpu}_autocast_${autocast}" + save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}" nodes=1 else IFS="," ips_array=(${ips}) IFS="|" nodes=${#ips_array[@]} - save_log="${LOG_PATH}/${model_name}/${trainer}_gpus_${gpu}_autocast_${autocast}_nodes_${nodes}" + save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}_nodes_${nodes}" fi -- GitLab