Update to remove "module" attribute check

5 years ago · ccfa9373ae
parent f83521ab9a
commit ccfa9373ae
1 changed files with 1 additions and 1 deletions
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@ -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 not in ["module", "process_group", "reducer"]:
+            if not k.startswith('_') and k not in ["process_group", "reducer"]:
                setattr(self.ema, k, v)