问题描述
是否可以在不导出的情况下将 PyTorch 模型转换为 ONNX,并在脚本中直接将其进一步用作 ONNX 对象。
解决方法
您可以导出到内存,如下所示:
import io
f = io.BytesIO()
torch.onnx.export(model,sample_inputs,f,...)
onnx_model = onnx.load_model_from_string(f.getvalue())
是否可以在不导出的情况下将 PyTorch 模型转换为 ONNX,并在脚本中直接将其进一步用作 ONNX 对象。
您可以导出到内存,如下所示:
import io
f = io.BytesIO()
torch.onnx.export(model,sample_inputs,f,...)
onnx_model = onnx.load_model_from_string(f.getvalue())