Skip to content

Commit

Permalink
gpu_id is None when args.use_gpu as False
Browse files Browse the repository at this point in the history
  • Loading branch information
LDOUBLEV committed Aug 17, 2021
1 parent 5bb6ae0 commit 49c4a3a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/infer/predict_det.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ def __init__(self, args):
inference_config=self.config,
pids=pid,
process_name=None,
gpu_ids=gpu_id,
gpu_ids=gpu_id if args.use_gpu else None,
time_keys=[
'preprocess_time', 'inference_time', 'postprocess_time'
],
Expand Down

0 comments on commit 49c4a3a

Please sign in to comment.