glenn-jocher
commited on
Commit
•
2b3109e
1
Parent(s):
2809616
Fix DDP destruction `LOGGER.info()` (#4863)
Browse files
train.py
CHANGED
@@ -510,7 +510,8 @@ def main(opt, callbacks=Callbacks()):
|
|
510 |
if not opt.evolve:
|
511 |
train(opt.hyp, opt, device, callbacks)
|
512 |
if WORLD_SIZE > 1 and RANK == 0:
|
513 |
-
|
|
|
514 |
|
515 |
# Evolve hyperparameters (optional)
|
516 |
else:
|
|
|
510 |
if not opt.evolve:
|
511 |
train(opt.hyp, opt, device, callbacks)
|
512 |
if WORLD_SIZE > 1 and RANK == 0:
|
513 |
+
LOGGER.info('Destroying process group... ')
|
514 |
+
dist.destroy_process_group()
|
515 |
|
516 |
# Evolve hyperparameters (optional)
|
517 |
else:
|