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()): ...@@ -527,9 +527,6 @@ def main(opt, callbacks=Callbacks()):
# Train # Train
if not opt.evolve: if not opt.evolve:
train(opt.hyp, opt, device, callbacks) 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) # Evolve hyperparameters (optional)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论