/Python/Export PyTorch model in the ONNX Runtime format

Export PyTorch model in the ONNX Runtime format

dummy_input = torch.randn(1, 3, 224, 224, device='cuda')
onnx_path =  "./model.onnx"
torch.onnx.export(learn.model, dummy_input, onnx_path, verbose=False)

# The output will be the model written to a file called model.onnx.

Similar codes

2 votes274 views2 answers
Python __repr__
Python
2 votes253 views1 answer
Python how to check which int var is the greatest
Python
0 vote293 views1 answer
Bouon arrondi tkinter
Python
0 vote274 views2 answers
Read image from a folder python
Python
0 vote272 views1 answer
FIND MISSING NUMBER IN AN ARRAY IN PYTHON
Python
0 vote294 views1 answer
Comment dériver une classe python
Python
2 votes280 views2 answers
Poetry add specific version
Python
0 vote260 views1 answer
How to insert files in tuple python
Python
-2 votes262 views1 answer
Python IOError
Python
0 vote280 views1 answer
Convert fisheye video to normal python
Python