Pytorch-lightning example script

Requirements

$ pip install pytorch-lightning
$ pip install pillow

Run

$ python train.py --image_folder /path/to/your/images