chore(core): remove redundant code in EMA (#396)

yolox/core/trainer.py

@@ -205,8 +205,6 @@ class Trainer:
                 self.save_ckpt(ckpt_name="last_mosaic_epoch")
 
     def after_epoch(self):
-        if self.use_model_ema:
-            self.ema_model.update_attr(self.model)
 
         self.save_ckpt(ckpt_name="latest")
 

yolox/utils/ema.py

@@ -20,15 +20,6 @@ def is_parallel(model):
     return isinstance(model, parallel_type)
 
 
-def copy_attr(a, b, include=(), exclude=()):
-    # Copy attributes from b to a, options to only include [...] and to exclude [...]
-    for k, v in b.__dict__.items():
-        if (len(include) and k not in include) or k.startswith("_") or k in exclude:
-            continue
-        else:
-            setattr(a, k, v)
-
-
 class ModelEMA:
     """
     Model Exponential Moving Average from https://github.com/rwightman/pytorch-image-models
@@ -68,7 +59,3 @@ class ModelEMA:
                 if v.dtype.is_floating_point:
                     v *= d
                     v += (1.0 - d) * msd[k].detach()
-
-    def update_attr(self, model, include=(), exclude=("process_group", "reducer")):
-        # Update EMA attributes
-        copy_attr(self.ema, model, include, exclude)