/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 vote325 views1 answer
How to comment a para in python
Python
2 votes361 views2 answers
Get query param in django
Python Django
2 votes302 views1 answer
Display all categorical column in pandas dataframe
Python
1 vote354 views1 answer
Django queryset multiple filters
Python Django
7 votes301 views2 answers
Python print for loop one line
Python
12 votes364 views1 answer
Python copy ddictionary
Python
28 votes329 views3 answers
Datetime one month ago python
Python
0 vote699 views1 answer
Warning message: In scan(file = file, what = what, sep = sep, quote = quote, dec = dec, : EOF within quoted string
Python
0 vote320 views1 answer
Renpy hide textbox
Python
0 vote298 views1 answer
Python mad libs
Python