mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

May resolve https://github.com/ultralytics/yolov5/issues/7307
pull/8957/head
Glenn Jocher 2022-08-13 03:57:31 +02:00 committed by GitHub
parent 11a76bb4f5
commit f1214f237d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions

3
train.py
View File

@ -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: