apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

EMA class updates

Browse Source
This commit is contained in:
Glenn Jocher 2020-03-14 18:08:48 -07:00
parent 851c9b9883
commit adba66c3a6
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

@ -151,4 +151,4 @@ class ModelEMA:
# Assign attributes (which may change during training) # Assign attributes (which may change during training)
for k in model.__dict__.keys(): for k in model.__dict__.keys():
if not k.startswith('_'): if not k.startswith('_'):
self.ema.__setattr__(k, model.getattr(k)) setattr(model, k, getattr(model, k))