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

Merge pull request #296 from NanoCode012/patch-1

Browse Source 
Fix save error for multi-gpu in ema
This commit is contained in:
Glenn Jocher 2020-07-04 11:50:34 -07:00 committed by GitHub
commit 9485f0b51f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/torch_utils.py
View File

@ -201,5 +201,5 @@ class ModelEMA:
def update_attr(self, model):
# Update EMA attributes
for k, v in model.__dict__.items():
if not k.startswith('_') and k != 'module':
if not k.startswith('_') and k not in ["process_group", "reducer"]:
setattr(self.ema, k, v)