Replicating the work of https://github.com/NadMag/LightLab

Currently only trains a UNet to memorize test data.

To use, edit file_paths.py to point at your dataset files.