Standalone tree split from LLMRL/projects/kda. Includes Triton dt_bias backward fix, train_k3 --preset 0.5b, SFT, Docker runtime, and tests.
6 lines
184 B
Python
6 lines
184 B
Python
"""Training and checkpoint helpers."""
|
|
|
|
from .toy import load_ckpt, make_toy_data, save_ckpt, train_one_batch
|
|
|
|
__all__ = ["load_ckpt", "make_toy_data", "save_ckpt", "train_one_batch"]
|