Remove DDP destroy_process_group() on train end (#8935)
May resolve https://github.com/ultralytics/yolov5/issues/7307
This commit is contained in:
1 parent
11a76bb4f5
commit
f1214f237d
1 file changed
-3
@@ -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:
|
||||||
|
|||||||
Reference in new issue
Block a user