Class Central is learner-supported. When you buy through links on our site, we may earn an affiliate commission.

YouTube

How to Convert Almost Any PyTorch Model to ONNX and Serve It Using Flask

Abhishek Thakur via YouTube

Overview

Learn how to convert a PyTorch BERT sentiment model to ONNX format and serve it using a Flask API in this 27-minute tutorial. Follow along as the process of converting the model, creating a new file, and setting up the API is demonstrated step-by-step. Discover the changes from previous videos, understand what components are no longer necessary, and explore the output of the conversion. Gain practical insights into deploying machine learning models for real-world applications using ONNX and Flask.

Syllabus

Introduction
Changes from previous videos
Converting model to ONNX
Creating a new file
What you dont need
Output
API

Taught by

Abhishek Thakur

Reviews

Start your review of How to Convert Almost Any PyTorch Model to ONNX and Serve It Using Flask

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Someone learning on their laptop while sitting on the floor.