训练 PyTorch 模型
PyTorch 是一个 Python 包,它为构建深度学习网络提供了 GPU 加速张量计算和高级功能。
MLflow PyTorch 笔记本在 MNIST 手写数字识别数据上安装了神经网络,并将运行结果记录到 MLflow 服务器。 它将训练指标和权重以 TensorFlow 事件格式记录在本地,然后将其上传到 MLflow 运行的项目目录。 最后,它启动 TensorBoard 并读取记录在本地的事件。
准备好后,可以使用 Azure Databricks 模型服务来部署模型。