pull/3272/head
LDOUBLEV 2021-07-06 15:12:15 +08:00
parent 10e3729fd5
commit 917118ae8e
1 changed files with 2 additions and 1 deletions

View File

@ -136,6 +136,7 @@ for gpu in ${gpu_list[*]}; do
env=""
elif [ ${#gpu} -le 1 ];then
env="export CUDA_VISIBLE_DEVICES=${gpu}"
eval ${env}
elif [ ${#gpu} -le 15 ];then
IFS=","
array=(${gpu})
@ -215,7 +216,7 @@ for gpu in ${gpu_list[*]}; do
status_check $? "${export_cmd}" "${status_log}"
#run inference
echo $env
eval $env
save_infer_path="${save_log}"
func_inference "${python}" "${inference_py}" "${save_infer_path}" "${LOG_PATH}" "${infer_img_dir}"
done