• Qiang Zhang's avatar
    Only import torch.distributed when needed (#882) · 8bf38df0
    Qiang Zhang authored
    * 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.
    8bf38df0
sampler.py 5.53 KB