LLM-GO / src /llm_go /training /__init__.py
LesterCerioli's picture
Building first version from Golang LLM
a58ece3
Raw
History Blame Contribute Delete
211 Bytes
"""Training loop, learning rate scheduling, and checkpointing."""
from llm_go.training.trainer import Trainer
from llm_go.training.lr_schedule import CosineWithWarmup
__all__ = ["Trainer", "CosineWithWarmup"]