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

Fix save error for multi-gpu

This commit is contained in:
NanoCode012 2020-07-04 16:50:33 +07:00 committed by GitHub
parent 5e2429e618
commit f83521ab9a
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 ["module", "process_group", "reducer"]:
setattr(self.ema, k, v) setattr(self.ema, k, v)