Click here to Skip to main content
15,075,266 members
Articles / Artificial Intelligence / Keras


4 bookmarked

Making PyTorch AI Models Portable Using ONNX

Rate me:
Please Sign up or sign in to vote.
5.00/5 (2 votes)
8 Sep 2020CPOL5 min read
In this article I provide a brief overview of PyTorch for those looking for a deep learning framework for building and training neural networks.
Here I then show how to convert PyTorch models to the ONNX format using the conversion tool which is already a part of PyTorch itself. I also show the best practice of adding metadata to the exported model.




This article, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)


About the Author

Keith Pijanowski
Technical Lead Bank of New York Mellon
United States United States
Keith is a sojourner in the software industry. He has over 30 years of experience building and bringing applications to market. He has worked for startups and large enterprises in roles ranging from tech lead to business development manager. He is currently a senior engineer on BNY Mellon's Distribution Analytics team where he is building data pipelines from on-premise data sources to the cloud.

Comments and Discussions