zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #296 from NanoCode012/patch-1 

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
parent 5e2429e618 ccfa9373ae
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): def update_attr(self, model):
# Update EMA attributes # Update EMA attributes
for k, v in model.__dict__.items(): 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) setattr(self.ema, k, v)