Pytorch 模型的批处理部署输入详细信息

在指定 Pytorch 模型的批处理部署的输入详细信息时，请遵循以下规则。

数据类型摘要表:

接受的数据类型和文件格式
数据 描述
类型 内联和数据引用
文件格式 包含 JSON 文件的 .zip 归档

数据源

输入或输出数据引用:

  • 空间中的本地资产或受管资产
  • 云存储中的联网（远程）资产

如果要以编程方式指定输入/输出数据引用：

  • 数据源引用 type 取决于资产类型。 请参阅 将数据资产添加到部署空间中的 数据源引用类型 部分。
  • 如果使用 ONNX 格式部署 Pytorch 模型，请指定 keep_initializers_as_inputs=True 标志并将 opset_version 设置为 9 (始终将 opset_version 设置为部署运行时支持的最新版本)。
    torch.onnx.export(net, x, 'lin_reg1.onnx', verbose=True, keep_initializers_as_inputs=True, opset_version=9)
注: 部署作业的环境变量参数不适用。