Объект ONNX из модели PyTorch без экспорта

Можно ли преобразовать модель PyTorch в ONNX без экспорта и в дальнейшем использовать ее как объект ONNX непосредственно в скрипте.


pytorch onnxruntime pytorch-lightning onnx onnx-coreml
person Sanjiban Sengupta    schedule 17.06.2021    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Вы можете экспортировать в память примерно так:

import io
f = io.BytesIO()
torch.onnx.export(model, sample_inputs, f, ...)
onnx_model = onnx.load_model_from_string(f.getvalue())
person Sergii Dymchenko    schedule 17.06.2021