def restore(self, state_dict, build_fp32_params=False):
        """Load data from a model spec into EMA model"""
        self.model.load_state_dict(state_dict, strict=False)
        if build_fp32_params:
            self.build_fp32_params(state_dict)
Load data from a model spec into EMA model
understand_icl/fairseq/fairseq/models/ema/ema.py