尧图网站设计 尧图网站设计YAOTU DESIGN
ARTICLE DETAIL

资讯详情

深耕网站设计与一线实操的经验洞察。

PyTorch实战(28)——PyTorch深度学习模型部署

PyTorch实战(28)——PyTorch深度学习模型部署 PyTorch实战(28)——PyTorch深度学习模型部署0. 前言1. 构建 PyTorch 模型推理管道1.1 保存和加载训练模型1.2 构建推理管道2. 构建模型服务器2.1 使用 Flask 编写基础应用2.2 使用 Flask 构建模型服务器3. 使用 Flask 服务器执行预测小结系列链接0. 前言我们已系统性地探讨了如何使用 PyTorch 训练和测试各类机器学习模型。我们从 PyTorch 的基础组件入手,掌握了高效完成深度学习任务的必备工具;随后深入研究了基于PyTorch实现的多种深度学习模型架构及其应用场景。在本节中,我们将重点讨论如何将这些模型投入生产环境。简单来说,就是讨论将一个已经训练并测试过的模型对象部署到一个独立的环境中,使其能够对新输入数据进行预测或推理。这称为模型的生产化,即将模型部署到生产系统中。本节我们将首先构建一个简易的PyTorch推理管道:通过输入数据和预训练模型的存储路径即可完成预测。随后将该推理管道部署至模型服务器,使其能够接收数据请求并返回预测结果。1. 构建 PyTorch 模型推理管道我们将使用在 PyTorch 深度学习概述一节中构建的MNIST手写数字分类模
返回列表