8b187bb
|
|
"""MiniMind Training Package"""
from .trainer import Mind2Trainer
from .distillation import DistillationTrainer
from .dataset import TextDataset, create_dataloader
__all__ = ["Mind2Trainer", "DistillationTrainer", "TextDataset", "create_dataloader"]
|
