Update to remove "module" attribute check
Browse files- utils/torch_utils.py +1 -1
utils/torch_utils.py
CHANGED
@@ -201,5 +201,5 @@ class ModelEMA:
|
|
201 |
def update_attr(self, model):
|
202 |
# Update EMA attributes
|
203 |
for k, v in model.__dict__.items():
|
204 |
-
if not k.startswith('_') and k not in ["
|
205 |
setattr(self.ema, k, v)
|
|
|
201 |
def update_attr(self, model):
|
202 |
# Update EMA attributes
|
203 |
for k, v in model.__dict__.items():
|
204 |
+
if not k.startswith('_') and k not in ["process_group", "reducer"]:
|
205 |
setattr(self.ema, k, v)
|