fix confilct

Evezerest · Nov 26, 2021 · b0e9694 · b0e9694
1 parent ddd07a6
commit b0e9694
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
@@ -118,6 +118,7 @@ if [ ${MODE} = "klquant_whole_infer" ] || [ ${MODE} = "system_infer" ] ; then
  image_dir_key=$(func_parser_key "${lines[16]}")
  infer_img_dir=$(func_parser_value "${lines[16]}")
  save_log_key=$(func_parser_key "${lines[17]}")
+ save_log_value=$(func_parser_value "${lines[17]}")
  benchmark_key=$(func_parser_key "${lines[18]}")
  benchmark_value=$(func_parser_value "${lines[18]}")
  infer_key1=$(func_parser_key "${lines[19]}")
@@ -161,8 +162,9 @@ function func_inference(){
  set_batchsize=$(func_set_params "${batch_size_key}" "${batch_size}")
  set_cpu_threads=$(func_set_params "${cpu_threads_key}" "${threads}")
  set_model_dir=$(func_set_params "${infer_model_key}" "${_model_dir}")
+ set_infer_params0=$(func_set_params "${save_log_key}" "${save_log_value}")
  set_infer_params1=$(func_set_params "${infer_key1}" "${infer_value1}")
- command="${_python} ${_script} ${use_gpu_key}=${use_gpu} ${use_mkldnn_key}=${use_mkldnn} ${set_cpu_threads} ${set_model_dir} ${set_batchsize} ${set_infer_data} ${set_benchmark} ${set_precision} ${set_infer_params1} > ${_save_log_path} 2>&1 "
+ command="${_python} ${_script} ${use_gpu_key}=${use_gpu} ${use_mkldnn_key}=${use_mkldnn} ${set_cpu_threads} ${set_model_dir} ${set_batchsize} ${set_infer_params0} ${set_infer_data} ${set_benchmark} ${set_precision} ${set_infer_params1} > ${_save_log_path} 2>&1 "
  eval $command
  last_status=${PIPESTATUS[0]}
  eval "cat ${_save_log_path}"