In this tutorial, see how you can train a Convolutional Neural Network in PyTorch and convert it into an ONNX model.
Once the model is in in ONNX format, you can import that into other frameworks such as TensorFlow for either inference and reusing the model through transfer learning.
This post is the third in a series of introductory tutorials on the Open Neural Network Exchange (ONNX), an initiative from AWS, Microsoft, and Facebook to define a standard for interoperability across machine learning platforms. See: Part 1 , Part 2 . In this tutorial, we will train a […]