Restore learning from intermediate state.

def restore(self, training_info: TrainingInfo, local_batch_idx: int, model: Model, hidden_state: dict):
        """
        Restore learning from intermediate state.
        """
        pass