use lerp in ema (#11442)

open-mmlab · Feb 1, 2024 · 498295a · 498295a
1 parent 76e23d2
commit 498295a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/models/layers/ema.py b/mmdet/models/layers/ema.py
@@ -63,4 +63,4 @@ def avg_func(self, averaged_param: Tensor, source_param: Tensor,
         """
         momentum = (1 - self.momentum) * math.exp(
             -float(1 + steps) / self.gamma) + self.momentum
-        averaged_param.mul_(1 - momentum).add_(source_param, alpha=momentum)
+        averaged_param.lerp_(source_param, momentum)