来自 PyTorch 模型的 ONNX 对象，无需导出

Posted 2023-03-27

技术标签:

【中文标题】来自 PyTorch 模型的 ONNX 对象，无需导出【英文标题】：ONNX object from PyTorch model without exporting 【发布时间】：2021-09-02 02:10:29 【问题描述】：

是否可以在不导出的情况下将 PyTorch 模型转换为 ONNX，并在脚本中直接将其用作 ONNX 对象。

【问题讨论】：

【参考方案1】：

你可以导出到内存，像这样：

import io
f = io.BytesIO()
torch.onnx.export(model, sample_inputs, f, ...)
onnx_model = onnx.load_model_from_string(f.getvalue())

【讨论】：

以上是关于来自 PyTorch 模型的 ONNX 对象，无需导出的主要内容，如果未能解决你的问题，请参考以下文章