INNER CODE UNIT · TypeScript
CheckpointCallback
theaniketgiri/create-llm · src/python-callback-templates.ts:75
class CheckpointCallback(Callback):
"""
Saves model checkpoints during training
Features:
- Save at regular intervals
- Keep only N most recent checkpoints
- Save best model based on validation loss
- Save on training interruption
"""
def __init__(
self,
checkpoint_dir: str = 'checkpoints',
save_interval: int = 1000,
save_total_limit: int = 3,
save_best: bool = True,
verbose: bool = True