Method reverse

funasr/models/data2vec/ema_module.py:136–146

Load the model parameters from EMA model. Useful for inference or fine-tuning from the EMA model.

(self, model)

        self._step_internal(new_model)

    def reverse(self, model):
        """
        Load the model parameters from EMA model.
        Useful for inference or fine-tuning from the EMA model.
        """
        d = self.model.state_dict()
        if "_ema" in d:
            del d["_ema"]

        model.load_state_dict(d, strict=False)
        return model