2025年onnx模型部署 python（onnx模型部署图和权重分开）

PyTorch 是用于训练深度学习模型的常用机器学习框架。在 Azure Databricks 中，PyTorch 预安装在 ML 群集中。

在 PyTorch 中，模型基于定义的网络。该网络由多个层组成，其中每个层都有指定的输入和输出。此外，该网络定义了一个 forward 函数，当数据通过网络传递时，该函数会将函数应用于每个层。

以下示例代码定义网络。

虽然代码乍一看似乎很复杂，但此类定义了一个相对简单的网络，其中包含三个层：

forward 函数将层应用于输入数据 (x)，并将每一层的输出传递到下一层，最后返回最后一层的输出（其中包含标签预测向量 y）。线性整流函数 (ReLU) 激活函数应用于第 1 层和第 2 层的输出，以将输出值限制为正数。

要创建用于训练的模型，只需创建网络类的实例，如下所示：

PyTorch 层处理格式为张量（类似矩阵的结构）的数据。有多种函数可以将其他常见数据格式转换为张量，你可以定义 PyTorch 数据加载程序，用于将数据张量读取到用于训练或推理的模型。

与大多数监督式机器学习方法一样，你应该为训练和验证定义单独的数据集。通过这种分离，可以验证模型在显示未经训练的数据时是否准确预测。

下面的代码定义了两个数据加载程序：一个用于训练，另一个用于测试。假定此示例中每个加载程序的源数据为 Numpy 特征值数组和 Numpy 对应标签值数组。

此示例中的加载程序将数据拆分为 30 个批，这些批在训练或推理过程中传递给 forward 函数。

通过将训练数据馈送到网络、度量损失（预测值和实际值之间的聚合差），并通过调整权重和平衡来优化网络，从而训练模型，以便将损失降到最低。关于如何计算和最小化损失的具体详细信息取决于所选的损失标准和优化器算法。

PyTorch 支持多种损失标准函数，包括（以及许多其他函数）：

若要指定训练模型时要使用的损失标准，可以创建相应函数的实例，如下所示：

计算出损失后，将使用优化器来确定如何最好地调整权重和平衡，以使其最小化。优化器是用于最小化函数的梯度下降方法的特定实现。 PyTorch 中可用的优化器包括（以及其他优化器）：

若要使用这些算法中的任何一种来训练模型，需要创建优化器的实例并设置任何所需的参数。具体参数因所选优化器而异，但大多数参数都要求指定用于控制对每次优化所做的调整大小的学习率。

以下代码创建 Adam 优化器的实例。

定义网络并为其准备数据后，可以使用数据来训练和测试模型，方法是通过网络传递训练数据、计算损失、优化网络权重和偏差，以及使用测试数据验证网络性能。通常的做法是，定义通过网络传递数据的函数以使用训练数据来训练模型，以及定义用于使用测试数据来测试模型的单独函数。

以下示例显示用于训练模型的函数。

以下示例显示用于测试模型的函数。

要训练深度学习模型，通常会多次运行训练函数（称为“时期”），目的是减少每个时期根据训练数据计算得出的损失。可以使用测试函数来验证根据测试数据（未训练模型）得出的损失是否也随着训练损失而减少，换句话说，模型训练不会生成过度拟合训练数据的模型。

以下代码训练模型超过 50 个时期。

成功训练模型后，可以保存其权重和偏差，如下所示：

要在以后加载和使用模型，请创建模型所基于的网络类的实例并加载保存的权重和偏差。

2025年onnx模型部署 python（onnx模型部署 图和权重分开）