Thus far, you have written your own implementation related to training and testing your machine learning model. However, much of the code you wrote was unrelated to your machine learning model architecture and could have applied to a broad range of distinct models. Building on this observation, this chapter introduces you to PyTorch Lightning, a framework that can help you reduce the amount of boilerplate engineering code in your machine learning system, and consequently help you focus on evolving your model design and implementation.
This section introduces the PyTorch Lightning framework for your PyTorch DC taxi fare estimation model and teaches you the steps involved in enabling PyTorch Lightning training, validation, and test features.