训练 PyTorch 模型

PyTorch 是一个 Python 包,它为构建深度学习网络提供了 GPU 加速张量计算和高级功能。

MLflow PyTorch 笔记本在 MNIST 手写数字识别数据上安装了神经网络,并将运行结果记录到 MLflow 服务器。 它将训练指标和权重以 TensorFlow 事件格式记录在本地,然后将其上传到 MLflow 运行的项目目录。 最后,它启动 TensorBoard 并读取记录在本地的事件。

准备好后,可以使用 Azure Databricks 模型服务来部署模型。

MLflow PyTorch 模型训练笔记本

获取笔记本