Skip to content

Commit

Permalink
Update test_coco_multiprocessing.py
Browse files Browse the repository at this point in the history
  • Loading branch information
yangxue0827 authored Nov 24, 2020
1 parent aab8650 commit bf05e25
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/test_coco_multiprocessing.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ def test_coco(det_net, real_test_img_list, eval_data, gpu_ids):
start = i * nr_image
end = min(start + nr_image, nr_records)
split_records = real_test_img_list[start:end]
proc = Process(target=worker, args=(i, split_records, det_net, eval_data, result_queue))
proc = Process(target=worker, args=(int(gpu_ids.strip().split(',')[i]), split_records, det_net, eval_data, result_queue))
print('process:%d, start:%d, end:%d' % (i, start, end))
proc.start()
procs.append(proc)
Expand Down

0 comments on commit bf05e25

Please sign in to comment.