def METHOD_NAME(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)