Book Image

PyTorch Deep Learning Hands-On

By : Sherin Thomas, Sudhanshu Passi
Book Image

PyTorch Deep Learning Hands-On

By: Sherin Thomas, Sudhanshu Passi

Overview of this book

PyTorch Deep Learning Hands-On is a book for engineers who want a fast-paced guide to doing deep learning work with PyTorch. It is not an academic textbook and does not try to teach deep learning principles. The book will help you most if you want to get your hands dirty and put PyTorch to work quickly. PyTorch Deep Learning Hands-On shows how to implement the major deep learning architectures in PyTorch. It covers neural networks, computer vision, CNNs, natural language processing (RNN), GANs, and reinforcement learning. You will also build deep learning workflows with the PyTorch framework, migrate models built in Python to highly efficient TorchScript, and deploy to production using the most sophisticated available tools. Each chapter focuses on a different area of deep learning. Chapters start with a refresher on how the model works, before sharing the code you need to implement it in PyTorch. This book is ideal if you want to rapidly add PyTorch to your deep learning toolset.
Table of Contents (11 chapters)
10
Index

Summary

In this chapter, we have gone through three different approaches to take PyTorch to production, starting from the easiest but least performant way: using Flask. Then we moved to the MXNet model server, which is a pre-built, optimized server implementation that can be managed using management APIs. The MXNet model server is useful for people who don't need a lot of complexity but need an efficient server implementation that can be scaled as required.

Lastly, we tried with TorchScript to create the most efficient version of our model and imported that in C++. For those who are ready to take up the complexity of building and maintaining a low-level language server like C++, Go, or Rust, you can take this approach and build a custom server until we have better runtime available that can read the script module and serve that like MXNet does on ONNX models.

The year 2018 was the year of model servers; there were numerous model servers from different organizations with...