跳到主要内容

用于深度学习的 MLflow

MLflow for ML Models 为深度学习工作流提供全面的实验追踪、模型管理和部署功能。从 PyTorch 训练循环到 TensorFlow 模型,MLflow 简化了您从实验到生产的路径。

快速开始

为何选择 MLflow 进行深度学习?

单行代码自动记录 (Autologging)

只需一行代码,即可为 PyTorch Lightning、TensorFlow 和 Keras 启用全面追踪。

实时监控

实时追踪跨 epoch 和批次的指标、损失曲线以及训练进度。

模型检查点

在训练过程中自动保存和版本化模型检查点,并提供完整的血缘(lineage)追踪。

生产部署

部署具备 GPU 加速、批处理推理和云平台集成的模型。

了解更多