Sample pytorch_lightning training code I developed for a project.