EMA class updates

2020-03-14 18:08:48 -07:00 · 2020-03-14 18:08:48 -07:00 · adba66c3a6
parent 851c9b9883
commit adba66c3a6
1 changed files with 1 additions and 1 deletions
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@ -151,4 +151,4 @@ class ModelEMA:
        # Assign attributes (which may change during training)
        for k in model.__dict__.keys():
            if not k.startswith('_'):
-                self.ema.__setattr__(k, model.getattr(k))
+                setattr(model, k, getattr(model, k))