Skip to content

Commit 96bb2dd

Browse files
authored
[Fix] Fix the redundant device_ids in tools/test.py. (#1215)
1 parent dc8691e commit 96bb2dd

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tools/test.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -195,10 +195,7 @@ def main():
195195
**show_kwargs)
196196
else:
197197
model = wrap_distributed_model(
198-
model,
199-
device=cfg.device,
200-
device_ids=[int(os.environ['LOCAL_RANK'])],
201-
broadcast_buffers=False)
198+
model, device=cfg.device, broadcast_buffers=False)
202199
outputs = multi_gpu_test(model, data_loader, args.tmpdir,
203200
args.gpu_collect)
204201

0 commit comments

Comments
 (0)