Pytorch Lightning: Creating My First Custom Data Module

How using a Pytorch Lightning eases things up. A. Lot. Pytorch is an excellent library if you are doing any projects or research in Deep Learning. However, if you want to do experiments, try different configurations of your network architectures, change your input data, and fiddle around with new loss functions; dealing with "boilerplate" code can be time-consuming and can make your code messy. This is the point where you might have thought about using Pytorch Lightning. If you have not, you might ought to, and I promise you it is worth it.