全面但轻量级深度学习训练框架,专注于你的模型、数据和评估!
BaseModel / BaseDataset / load_cfg 可独立于训练栈使用;需要编排时再与 Runner/Loops、回调、评估自由组合。 查看 API 概览。
单卡/多卡/DeepSpeed 一体化支持,fp16、梯度累计、调度器 与 回调/检查点/日志 开箱即用。 参见 Runner 与 Loops。
BaseMapDataset / BaseIterableDataset 统一 map/filter 与批处理; 评估使用 Evaluator + BaseMetric 组合。参见 Dataset 与 Eval。
map
filter