Example CNN written using Pytorch Lightning. - View it on GitHub