如何解决来自 PyTorch 模型的 ONNX 对象,无需导出
是否可以在不导出的情况下将 PyTorch 模型转换为 ONNX,并在脚本中直接将其进一步用作 ONNX 对象。
解决方法
您可以导出到内存,如下所示:
import io
f = io.BytesIO()
torch.onnx.export(model,sample_inputs,f,...)
onnx_model = onnx.load_model_from_string(f.getvalue())
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。