https://blog.xperience.ai/introduction-to-pytorch-lightning/