diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh index 2b1ca65a36972792d579f0a784303e5652f427cc..7d70de1e38d2eb1a4c27e12eeea11d8c3d76975b 100644 --- a/test_tipc/test_train_inference_python.sh +++ b/test_tipc/test_train_inference_python.sh @@ -103,7 +103,7 @@ if [ ${MODE} = "klquant_whole_infer" ]; then # parser inference inference_py=$(func_parser_value "${lines[8]}") use_gpu_key=$(func_parser_key "${lines[9]}") - use_gpu_list=$(func_parser_value "${lines[10]}") + use_gpu_list=$(func_parser_value "${lines[9]}") use_mkldnn_key=$(func_parser_key "${lines[10]}") use_mkldnn_list=$(func_parser_value "${lines[10]}") cpu_threads_key=$(func_parser_key "${lines[11]}")