PyTorch Lightning implementation

[lcoandrade]

I made a version in PyTorch Lightning (based on your work) using a diferente Patch Discriminator. My results tend to be more "conservative" than yours. Your implementation outputs more colorful results. I would like to know if you can check my code to see were I can improve. Is that possible? My code is available on Kaggle (https://www.kaggle.com/code/luizclaudioandrade/b-w-colorizing-pytorch-lightning/notebook).

Thanks in advance!