def METHOD_NAME(self, model: nn.Module, state_dict_increase: Dict[str, Any]):
    '''
    The final reconstruction step
    '''
    # naive approach
    model.load_state_dict({k: v + model.state_dict()[k] for k, v in state_dict_increase.items()}, strict=False)