跳到主要内容

MLflow 深度学习

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

开始使用

为什么选择 MLflow 进行深度学习?

单行自动日志记录

通过 PyTorch Lightning、TensorFlow 和 Keras 的单行代码启用全面的跟踪。

实时监控

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

模型检查点

在训练过程中自动保存和版本化模型检查点,并具有完整的谱系跟踪。

生产部署

使用 GPU 加速、批处理推理和云平台集成部署模型。

了解更多