Skip to content
Snippets Groups Projects
Forked from nikhil_rayaprolu / food-round2
Source project has a limited visibility.
user avatar
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
History
Name Last commit Last update
..