跳到主要内容

MLflow for Deep Learning

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

开始使用

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

单行自动日志记录

通过为 PyTorch Lightning、TensorFlow 和 Keras 编写一行代码,即可实现全面的跟踪。

实时监控

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

模型检查点

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

生产部署

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

了解更多