Search icon
Arrow left icon
All Products
Best Sellers
New Releases
Books
Videos
Audiobooks
Learning Hub
Newsletters
Free Learning
Arrow right icon
Accelerate Model Training with PyTorch 2.X

You're reading from  Accelerate Model Training with PyTorch 2.X

Product type Book
Published in Apr 2024
Publisher Packt
ISBN-13 9781805120100
Pages 230 pages
Edition 1st Edition
Languages
Author (1):
Maicon Melo Alves Maicon Melo Alves
Profile icon Maicon Melo Alves

Table of Contents (17) Chapters

Preface Part 1: Paving the Way
Chapter 1: Deconstructing the Training Process Chapter 2: Training Models Faster Part 2: Going Faster
Chapter 3: Compiling the Model Chapter 4: Using Specialized Libraries Chapter 5: Building an Efficient Data Pipeline Chapter 6: Simplifying the Model Chapter 7: Adopting Mixed Precision Part 3: Going Distributed
Chapter 8: Distributed Training at a Glance Chapter 9: Training with Multiple CPUs Chapter 10: Training with Multiple GPUs Chapter 11: Training with Multiple Machines Index Other Books You May Enjoy

Why do we need an efficient data pipeline?

We’ll start this chapter by making you aware of the relevance of having an efficient data pipeline. In the next few subsections, you will understand what a data pipeline is and how it can impact the performance of the training process.

What is a data pipeline?

As you learned in Chapter 1, Deconstructing the Training Process, the training process is composed of four phases: forward, loss calculation, optimization, and backward. The training algorithm iterates on dataset samples until there’s a complete epoch. Nevertheless, there is an additional phase we excluded from that explanation: data loading.

The forward phase invokes data loading to get dataset samples to execute the training process. More specifically, the forward phase calls the data loading process on each iteration to get the data required to execute the current training step, as shown in Figure 5.1:

Figure 5.1 – Data loading process

Figure 5.1 – Data loading...

lock icon The rest of the chapter is locked
Register for a free Packt account to unlock a world of extra content!
A free Packt account unlocks extra newsletters, articles, discounted offers, and much more. Start advancing your knowledge today.
Unlock this book and the full library FREE for 7 days
Get unlimited access to 7000+ expert-authored eBooks and videos courses covering every tech area you can think of
Renews at $15.99/month. Cancel anytime}