/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

0 vote189 views1 answer
How to connect wifi using python
Python
3 votes205 views1 answer
Last history of whatsapp message with python
Python
0 vote211 views6 answers
Gdal user with anaconda
Python
0 vote197 views1 answer
Pyton get minimum value of array
Python
2 votes184 views1 answer
How to add a function in python
Python
9 votes191 views3 answers
Make zipfile from directory py
Python
2 votes166 views1 answer
Matplotlib insert text
Python
0 vote195 views1 answer
Python array remove new element
Python
9 votes218 views2 answers
Argc python
Python
-1 vote177 views2 answers
Python how to insert values into string
Python