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: • Training Sentiment Model Using BERT and Se...
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: / abhi1thakur
LinkedIn: / abhi1thakur
Kaggle: https://kaggle.com/abhishek