/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

3 votes400 views6 answers
Django login
Python Django
38 votes345 views2 answers
Python extract thefile name from relative path
Python
0 vote344 views1 answer
Python override add
Python
2 votes323 views2 answers
Int' object is not subscriptable in python
Python
47 votes354 views4 answers
Sorting a dictionary by value in python
Python
1 vote373 views1 answer
How to save inputs python
Python
4 votes402 views4 answers
How recursion works in python
Python
0 vote341 views1 answer
How to add a under line of the button in tkinter python
Python
1 vote357 views1 answer
How to use if else to prove a variable even or odd in python
Python
1 vote368 views2 answers
How to add pagination in discord.py
Python