[Fix] Fix ddp ut (#2060)

open-mmlab · Sep 13, 2022 · ca7c098 · ca7c098
1 parent b51670b
commit ca7c098
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_utils/test_util_distribution.py b/tests/test_utils/test_util_distribution.py
@@ -56,7 +56,7 @@ def test_build_ddp():
 
  if torch.cuda.is_available():
  mmddp = build_ddp(
- model, 'cuda', device_id=[0], process_group=MagicMock())
+ model, 'cuda', device_ids=[0], process_group=MagicMock())
  assert isinstance(mmddp, MMDistributedDataParallel)
 
  if digit_version(mmcv.__version__) >= digit_version('1.5.0'):