INNER CODE UNIT · TypeScript

on_batch_end

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

    def on_batch_end(self, trainer: Any, batch: Dict[str, Any], outputs: Dict[str, Any]):
        """Called after processing each batch"""
        pass
    
    def on_step_end(self, trainer: Any, step: int, loss: float, metrics: Optional[Dict[str, float]] = None):
        """Called after each training step"""
        pass
`;
  }

  /**
   * Get CheckpointCallback
   */
  static getCheckpointCallback(): string {
    return `"""
Checkpoint callback for saving model checkpoints during training
"""

View source record →