From e0520cf7e9a36748a460de5808d27900d5ab0d12 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 5 Aug 2021 07:34:33 +0000 Subject: [PATCH] fix --- tests/test.sh | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/test.sh b/tests/test.sh index c57532fd..962ff50a 100644 --- a/tests/test.sh +++ b/tests/test.sh @@ -235,17 +235,20 @@ if [ ${MODE} = "infer" ]; then # run export if [ ${infer_run_exports[Count]} != "null" ];then set_export_weight=$(func_set_params "${export_weight}" "${infer_model}") - set_save_infer_key=$(func_set_params "${save_infer_key}" "${infer_model}") + set_save_infer_key=$(func_set_params "${save_infer_key}" "${infer_model}_infer") export_cmd="${python} ${norm_export} ${set_export_weight} ${set_save_infer_key}" eval $export_cmd status_export=$? if [ ${status_export} = 0 ];then status_check $status_export "${export_cmd}" "${status_log}" fi + load_infer_model="${infer_model}_infer" + else + load_infer_model="${infer_model}" fi #run inference is_quant=${infer_quant_flag[Count]} - func_inference "${python}" "${inference_py}" "${infer_model}" "${LOG_PATH}" "${infer_img_dir}" ${is_quant} + func_inference "${python}" "${inference_py}" "${load_infer_model}" "${LOG_PATH}" "${infer_img_dir}" ${is_quant} Count=$(($Count + 1)) done -- GitLab