mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Remove DDP destroy_process_group() on train end (#8935)

Browse Source 
May resolve https://github.com/ultralytics/yolov5/issues/7307
This commit is contained in:
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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: