Photo by SpaceX on Unsplash

Deploy models in PyTorch 🚀

torchserve to the rescue!

All the code used in this article is here

Recently, PyTorch has introduced its new production framework to properly serve models, called torchserve.So, without further due, let’s present today’s roadmap:

1. Installation with Docker
2. Export your model
3. Define a handler