7c29567dc5f898507530d71003bdf80eaab0fbb2,orbit/controller.py,Controller,__init__,#Controller#Any#Any#Any#Any#Any#Any#Any#Any#Any#,36

Before Change

    if self.checkpoint_manager is not None:
      checkpoint_interval = self.checkpoint_manager.checkpoint_interval
      model_restored = self.restore_checkpoint()
      if not model_restored and (checkpoint_interval and
                                 self.trainer is not None):
        // If the model is not restored from a checkpoint, and
        // `checkpoint_interval` is enabled for training, save an initial
        // checkpoint.
        self.save_checkpoint()

  def train(self, steps: int, checkpoint_at_completion: bool = True):
    Runs training.

    This method calls the `train` method on the Trainable object until the

After Change

    if self.checkpoint_manager is not None:
      checkpoint_interval = self.checkpoint_manager.checkpoint_interval
      restored_path = self.restore_checkpoint()
      if restored_path:
        logging.info("Restored from checkpoint: %s", restored_path)

  def train(self, steps: int, checkpoint_at_completion: bool = True):
    Runs training.

    This method calls the `train` method on the Trainable object until the

Instances