Skip to content

Commit bf05e25

Browse files
authored
Update test_coco_multiprocessing.py
1 parent aab8650 commit bf05e25

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/test_coco_multiprocessing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def test_coco(det_net, real_test_img_list, eval_data, gpu_ids):
9595
start = i * nr_image
9696
end = min(start + nr_image, nr_records)
9797
split_records = real_test_img_list[start:end]
98-
proc = Process(target=worker, args=(i, split_records, det_net, eval_data, result_queue))
98+
proc = Process(target=worker, args=(int(gpu_ids.strip().split(',')[i]), split_records, det_net, eval_data, result_queue))
9999
print('process:%d, start:%d, end:%d' % (i, start, end))
100100
proc.start()
101101
procs.append(proc)

0 commit comments

Comments
 (0)