From ff62587221fd8bc4f3fcabdc3754dbc0ccc6f1cb Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Wed, 19 Jan 2022 06:59:46 +0000
Subject: [PATCH] opt train log

---
 test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt | 4 ++++
 test_tipc/test_train_inference_python.sh              | 4 ++--
 2 files changed, 6 insertions(+), 2 deletions(-)

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 7942c7cf..d8829297 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 c9ca2d50..e6f10e0d 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
 
 
-- 
GitLab