File size: 251 Bytes
8b187bb
 
 
 
 
 
1
2
3
4
5
6
7
"""MiniMind Training Package"""
from .trainer import Mind2Trainer
from .distillation import DistillationTrainer
from .dataset import TextDataset, create_dataloader

__all__ = ["Mind2Trainer", "DistillationTrainer", "TextDataset", "create_dataloader"]