Parallelizing Neural Network Training with PyTorch
In this chapter, we will move on from the mathematical foundations of machine learning and deep learning to focus on PyTorch. PyTorch is one of the most popular deep learning libraries currently available, and it lets us implement neural networks (NNs) much more efficiently than any of our previous NumPy implementations. In this chapter, we will start using PyTorch and see how it brings significant benefits to training performance.
This chapter will begin the next stage of our journey into machine learning and deep learning, and we will explore the following topics:
- How PyTorch improves training performance
- Working with PyTorch’s
Dataset
andDataLoader
to build input pipelines and enable efficient model training - Working with PyTorch to write optimized machine learning code
- Using the
torch.nn
module to implement common deep learning architectures conveniently - Choosing activation functions...