Commit e74c260f authored by Kai Chen's avatar Kai Chen
Browse files

fix typo

parent 9569fb5b
......@@ -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))
......
Markdown is supported
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