/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 vote109 views1 answer
Non blocking socket python
Python
0 vote111 views1 answer
Pandas, python, dataframes
Python
0 vote96 views2 answers
Which columns has null data?
Python
0 vote95 views1 answer
Python pass string with spaces as argument
Python
0 vote95 views1 answer
Bitbucket rest api python example
Python
0 vote99 views1 answer
Boto3 read excel file from s3
Python
0 vote135 views1 answer
Python scripting
Python
1 vote115 views1 answer
Special characters list in python
Python
2 votes116 views2 answers
Pandas set options
Python
0 vote110 views1 answer
How to read comment before the root element of xml python
Python