def METHOD_NAME(self, model_parameters):
    if self.optimizer is None:
        return
    current_opt_state = self.optimizer.init(model_parameters)
    self.aggregator_state = TrainingState(model_parameters, model_parameters, current_opt_state)