diff --git a/mmdet/core/utils/dist_utils.py b/mmdet/core/utils/dist_utils.py index 2c0a9283964f4b27e7ec59feaa825a14deaf9ffc..7ffa7a093481098e937cfa1b287b57980dd3d185 100644 --- a/mmdet/core/utils/dist_utils.py +++ b/mmdet/core/utils/dist_utils.py @@ -19,9 +19,9 @@ def init_dist(launcher, backend='nccl', **kwargs): if launcher == 'pytorch': _init_dist_pytorch(backend, **kwargs) elif launcher == 'mpi': - _init_dist_pytorch(backend, **kwargs) + _init_dist_mpi(backend, **kwargs) elif launcher == 'slurm': - _init_dist_pytorch(backend, **kwargs) + _init_dist_slurm(backend, **kwargs) else: raise ValueError('Invalid launcher type: {}'.format(launcher))