Pytorch lightning playground

Reimplementation of a simple Convolutional neural network using features of Pytorch lightning in purpose of getting used to this framework