Go back to the English version of the documentationPytorch 模型的批处理部署输入详细信息
Pytorch 模型的批处理部署输入详细信息
Last updated: 2024年10月07日
在指定 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)
注: 部署作业的环境变量参数不适用。
父主题: 按框架列出的批处理部署输入详细信息