I tried training a UNet model written in pytorch but i cant seem to make it work. I tried training on a single image (the dataset is Carvana) for 500 epochs but the output is pure black. Any help would be appreciated.
Here is the link to my Kaggle kernel: Carvana-Pytorch 6