Unverified 提交 f1214f23 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Remove DDP `destroy_process_group()` on train end (#8935)

上级 11a76bb4
......@@ -527,9 +527,6 @@ def main(opt, callbacks=Callbacks()):
# Train
if not opt.evolve:
train(opt.hyp, opt, device, callbacks)
if WORLD_SIZE > 1 and RANK == 0:
LOGGER.info('Destroying process group... ')
dist.destroy_process_group()
# Evolve hyperparameters (optional)
else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论