/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

1 vote149 views2 answers
Get required packages from python project
Python
49 votes152 views6 answers
Sort dictionary
Python
0 vote145 views1 answer
Django multi column index
Python
3 votes150 views2 answers
Secondary y axis matplotlib
Python
2 votes135 views3 answers
Python seaborn color map
Python
4 votes189 views1 answer
Skip to next iteration in for loop python
Python
0 vote150 views1 answer
Plt.imread python
Python
0 vote198 views1 answer
Get hash python
Python
0 vote196 views1 answer
Django button
Python Django
0 vote157 views1 answer
Distribute a local Python collection to form an RDD
Python