Method update

ppocr/utils/ema.py:84–94

Update shadow weights with current model parameters.

(self, model)

            return self.decay

    def update(self, model):
        """Update shadow weights with current model parameters."""
        decay = self._get_decay()
        self._decay = decay
        model_dict = model.state_dict()
        for k, v in self.state_dict.items():
            if k not in self.ema_black_list and k in model_dict:
                v = decay * v + (1 - decay) * model_dict[k].astype("float32")
                v.stop_gradient = True
                self.state_dict[k] = v
        self.step += 1

    def apply(self):
        """Return bias-corrected EMA state dict for eval/save.