2d34259281a6251cbdc67103ce8a1310010b8ceb,orbit/controller.py,Controller,__init__,#Controller#,39

Before Change

      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.

After Change

    if self.checkpoint_manager is not None:
      restored_path = self.restore_checkpoint()
      if restored_path:
        _log(f"restored from checkpoint: {restored_path}")

  def train(self, steps: int, checkpoint_at_completion: bool = True):
    Runs training until the specified global step count has been reached.

Instances