/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 vote333 views1 answer
Python launch ipython from script
Python
0 vote332 views1 answer
Apyori
Python
0 vote343 views1 answer
Search bar python program
Python
1 vote287 views1 answer
Python twilio certificate error
Python
0 vote292 views1 answer
DataFrame changed by function
Python
0 vote289 views1 answer
Fibonacci list pyton
Python
6 votes282 views2 answers
Python min in dictionary
Python
1 vote305 views1 answer
Python fill table wiget
Python
0 vote277 views1 answer
How to read an xml file
Python
1 vote343 views1 answer
Calculator in one line in python
Python