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 
edgelm/fairseq/models/ema/ema.py