torch icon indicating copy to clipboard operation
torch copied to clipboard

model to ONNX

Open VladPerervenko opened this issue 2 years ago • 4 comments

Is it possible to add a model converter to ONNX ?

VladPerervenko avatar Mar 01 '23 19:03 VladPerervenko

Native PyTorch 1.12-1.13 have a torch.onnx.export() method. Can this method be ported to Torch(R)? Or in another way to convert the model trained in R to ONNH?

VladPerervenko avatar Mar 25 '23 09:03 VladPerervenko

Unfortunatelly there's currently there's no way of doing it directly from R. You can potentially jit_trace your model, load it in PyTorch and export it to ONNX.

dfalbel avatar Mar 25 '23 09:03 dfalbel

Thank you. I will check the option

VladPerervenko avatar Mar 25 '23 10:03 VladPerervenko

Is there any interest in supporting an ONNX export feature in the future?

Exportable/Transportable/Sharable models are pretty important/relevant right now

tech-team-rural-mda avatar Mar 21 '24 19:03 tech-team-rural-mda