训练 PyTorch 模型

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

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

MLflow PyTorch 模型训练笔记本

获取笔记本