Skip to content

GitLab

Commit d6eb34f6 authored by littletomatodonkey's avatar littletomatodonkey
Browse files

fix infer

parent 916cec34
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tools/infer/predict_system.py
View file @ d6eb34f6
...@@ -188,12 +188,13 @@ def main(args): ...@@ -188,12 +188,13 @@ def main(args):
if __name__ == "__main__": if __name__ == "__main__":
args = utility.parse_args() args = utility.parse_args()
if args.use_mp: if args.use_mp:
print("12233")
p_list = [] p_list = []
total_process_num = args.total_process_num total_process_num = args.total_process_num
for process_id in range(total_process_num): for process_id in range(total_process_num):
cmd = [sys.executable, "-u"] + sys.argv + [ cmd = [sys.executable, "-u"] + sys.argv + [
"--process_id={} ".format(process_id), "--process_id={}".format(process_id),
"--use_mp={} ".format(args.use_mp) "--use_mp={}".format(False)
] ]
p = subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stdout) p = subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stdout)
p_list.append(p) p_list.append(p)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment