提交 908ee794 编写于 作者: A andyjpaddle

update mode

上级 a4f44542
...@@ -11,6 +11,7 @@ function func_parser_model_config(){ ...@@ -11,6 +11,7 @@ function func_parser_model_config(){
FILENAME=$1 FILENAME=$1
dataline=$(awk 'NR==1, NR==19{print}' $FILENAME) dataline=$(awk 'NR==1, NR==19{print}' $FILENAME)
MODE=$2
# parser params # parser params
IFS=$'\n' IFS=$'\n'
...@@ -51,7 +52,7 @@ cpp_client_py=$(func_parser_value "${lines[17]}") ...@@ -51,7 +52,7 @@ cpp_client_py=$(func_parser_value "${lines[17]}")
image_dir_key=$(func_parser_key "${lines[18]}") image_dir_key=$(func_parser_key "${lines[18]}")
image_dir_value=$(func_parser_value "${lines[18]}") image_dir_value=$(func_parser_value "${lines[18]}")
LOG_PATH="$(pwd)/test_tipc/output/${model_name}/cpp_serving" LOG_PATH="$(pwd)/test_tipc/output/${model_name}/${MODE}/cpp"
mkdir -p ${LOG_PATH} mkdir -p ${LOG_PATH}
status_log="${LOG_PATH}/results_cpp_serving.log" status_log="${LOG_PATH}/results_cpp_serving.log"
...@@ -167,7 +168,7 @@ function func_serving(){ ...@@ -167,7 +168,7 @@ function func_serving(){
#set cuda device #set cuda device
GPUID=$2 GPUID=$3
if [ ${#GPUID} -le 0 ];then if [ ${#GPUID} -le 0 ];then
env="export CUDA_VISIBLE_DEVICES=0" env="export CUDA_VISIBLE_DEVICES=0"
else else
......
...@@ -11,6 +11,7 @@ function func_parser_model_config(){ ...@@ -11,6 +11,7 @@ function func_parser_model_config(){
FILENAME=$1 FILENAME=$1
dataline=$(awk 'NR==1, NR==23{print}' $FILENAME) dataline=$(awk 'NR==1, NR==23{print}' $FILENAME)
MODE=$2
# parser params # parser params
IFS=$'\n' IFS=$'\n'
...@@ -58,7 +59,7 @@ pipeline_py=$(func_parser_value "${lines[21]}") ...@@ -58,7 +59,7 @@ pipeline_py=$(func_parser_value "${lines[21]}")
image_dir_key=$(func_parser_key "${lines[22]}") image_dir_key=$(func_parser_key "${lines[22]}")
image_dir_value=$(func_parser_value "${lines[22]}") image_dir_value=$(func_parser_value "${lines[22]}")
LOG_PATH="$(pwd)/test_tipc/output/${model_name}/python_serving" LOG_PATH="$(pwd)/test_tipc/output/${model_name}/${MODE}/python"
mkdir -p ${LOG_PATH} mkdir -p ${LOG_PATH}
status_log="${LOG_PATH}/results_python_serving.log" status_log="${LOG_PATH}/results_python_serving.log"
...@@ -207,7 +208,7 @@ function func_serving(){ ...@@ -207,7 +208,7 @@ function func_serving(){
#set cuda device #set cuda device
GPUID=$2 GPUID=$3
if [ ${#GPUID} -le 0 ];then if [ ${#GPUID} -le 0 ];then
env="export CUDA_VISIBLE_DEVICES=0" env="export CUDA_VISIBLE_DEVICES=0"
else else
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册