跳到主要内容

Wall E | 瓦力

全面但轻量级深度学习训练框架,专注于你的模型、数据和评估!

解耦组件,可单独使用

BaseModel / BaseDataset / load_cfg 可独立于训练栈使用;需要编排时再与 Runner/Loops、回调、评估自由组合。 查看 API 概览

训练编排:Runner 与 Loops

单卡/多卡/DeepSpeed 一体化支持,fp16、梯度累计、调度器回调/检查点/日志 开箱即用。 参见 RunnerLoops

数据流水线与评估

BaseMapDataset / BaseIterableDataset 统一 map/filter 与批处理; 评估使用 Evaluator + BaseMetric 组合。参见 DatasetEval