INNER CODE UNIT · TypeScript

on_step_end

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

    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
"""

import torch
from pathlib import Path
from typing import Any, Optional
from .base import Callback

View source record →