Skip to content
Snippets Groups Projects
Commit 8bf38df0 authored by Qiang Zhang's avatar Qiang Zhang Committed by Kai Chen
Browse files

Only import torch.distributed when needed (#882) 

* Fix an import error for `get_world_size` and `get_rank`

* Only import torch.distributed when needed

torch.distributed is only used in DistributedGroupSampler

* use `get_dist_info` to obtain world size and rank

`get_dist_info` from `mmcv.runner.utils` handles the problem of `distributed_c10d` doesn't exist.
parent f080ccbe
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment