How to convert almost any PyTorch model to ONNX and serve it using flask

How to convert almost any PyTorch model to ONNX and serve it using flask

25.619 Lượt nghe
How to convert almost any PyTorch model to ONNX and serve it using flask
In this video, I show you how you can convert any #PyTorch model to #ONNX format and serve it using flask api. I will be converting the #BERT sentiment model that we built in previous videos. Training Sentiment Model Using BERT and Serving it with Flask APl: https://www.youtube.com/watch?v=hinZO--TEk4 Please subscribe and like the video to help me keep motivated to make awesome videos like this one. :) To buy my book, Approaching (Almost) Any Machine Learning problem, please visit: https://bit.ly/buyaaml Follow me on: Twitter: https://twitter.com/abhi1thakur LinkedIn: https://www.linkedin.com/in/abhi1thakur/ Kaggle: https://kaggle.com/abhishek