0 / 0
Go back to the English version of the documentation
Pytorch 模型的批处理部署输入详细信息
Last updated: 2024年10月07日
Pytorch 模型的批处理部署输入详细信息

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

数据类型摘要表:

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

数据源

输入或输出数据引用:

  • 空间中的本地资产或受管资产
  • 已连接(远程)资产:Cloud Object Storage

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

  • 数据源引用 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)
    
注: 部署作业的环境变量参数不适用。

父主题: 按框架列出的批处理部署输入详细信息

Generative AI search and answer
These answers are generated by a large language model in watsonx.ai based on content from the product documentation. Learn more