INNER CODE UNIT · TypeScript
__init__
theaniketgiri/create-llm · src/python-callback-templates.ts:86
def __init__(
self,
checkpoint_dir: str = 'checkpoints',
save_interval: int = 1000,
save_total_limit: int = 3,
save_best: bool = True,
verbose: bool = True
):
"""
Initialize checkpoint callback
Args:
checkpoint_dir: Directory to save checkpoints
save_interval: Save checkpoint every N steps
save_total_limit: Maximum number of checkpoints to keep
save_best: Whether to save best model
verbose: Print checkpoint messages
"""