INNER CODE UNIT · TypeScript
on_train_end
theaniketgiri/create-llm · src/python-callback-templates.ts:35
def on_train_end(self, trainer: Any):
"""Called when training ends"""
pass
def on_epoch_begin(self, trainer: Any, epoch: int):
"""Called at the beginning of each epoch"""
pass
def on_epoch_end(self, trainer: Any, epoch: int):
"""Called at the end of each epoch"""
pass
def on_batch_begin(self, trainer: Any, batch: Dict[str, Any]):
"""Called before processing each batch"""
pass
def on_batch_end(self, trainer: Any, batch: Dict[str, Any], outputs: Dict[str, Any]):
"""Called after processing each batch"""