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

Fix DDP destruction `LOGGER.info()` (#4863)

上级 28096163
...@@ -510,7 +510,8 @@ def main(opt, callbacks=Callbacks()): ...@@ -510,7 +510,8 @@ def main(opt, callbacks=Callbacks()):
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: if WORLD_SIZE > 1 and RANK == 0:
_ = LOGGER.info('Destroying process group... ', end=''), dist.destroy_process_group(), LOGGER.info('Done.') LOGGER.info('Destroying process group... ')
dist.destroy_process_group()
# Evolve hyperparameters (optional) # Evolve hyperparameters (optional)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论