From 498295aad4e7081a1b8540c299c3ff9ff9a01956 Mon Sep 17 00:00:00 2001 From: RangiLyu Date: Thu, 1 Feb 2024 10:51:51 +0800 Subject: [PATCH] use `lerp` in ema (#11442) --- mmdet/models/layers/ema.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdet/models/layers/ema.py b/mmdet/models/layers/ema.py index bce503c4641..73a0ca67c28 100644 --- 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)