INNER CODE UNIT · TypeScript

on_step_end

theaniketgiri/create-llm · src/python-callback-templates.ts:120

    def on_step_end(self, trainer: Any, step: int, loss: float, metrics: Optional[dict] = None):
        """Save checkpoint at intervals"""
        if step % self.save_interval == 0 and step > 0:
            self._save_checkpoint(trainer, step, loss, is_best=False)
    
    def on_train_end(self, trainer: Any):
        """Save final checkpoint"""
        self._save_checkpoint(trainer, trainer.global_step, trainer.last_loss, is_final=True)
    
    def _save_checkpoint(
        self,
        trainer: Any,
        step: int,
        loss: float,
        is_best: bool = False,
        is_final: bool = False
    ):
        """Save a checkpoint"""

View source record →