diff --git a/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt b/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt index 7942c7cfe2bdbb8ce8956c1dee6f7bf5f3bd27fd..d8829297a3861c430482c18148cb65c89c960d3c 100644 --- a/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt +++ b/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt @@ -24,3 +24,7 @@ null:null eval:null null:null ## +===========================infer_params=========================== +Global.save_inference_dir:./output/ +Global.checkpoints: +norm_export:null diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh index c9ca2d504fe34e1b5f2eae91cdce208d7d0ff150..e6f10e0da69f1592e03043f3c0d4a815f5e29abc 100644 --- a/test_tipc/test_train_inference_python.sh +++ b/test_tipc/test_train_inference_python.sh @@ -323,14 +323,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}/${trainer}_gpus_${gpu}_autocast_${autocast}" + save_log="${LOG_PATH}/${model_name}_gpus_${gpu}_autocast_${autocast}_bs_${train_batch_value}_sp" nodes=1 else IFS="," ips_array=(${ips}) IFS="|" nodes=${#ips_array[@]} - save_log="${LOG_PATH}/${trainer}_gpus_${gpu}_autocast_${autocast}_nodes_${nodes}" + save_log="${LOG_PATH}/${model_name}_gpus_${gpu}_autocast_${autocast}_bs_${train_batch_value}_mp" fi