The model only trains for a few seconds because I’m only using 1 training image and testing with 1 image. I know that’s nothing but I think we should still see some reasonable but garbage output right? Especially since I am using a UNET for multi-class segmentation, I thought one of the key features of UNETs are their ability to learn quickly with small data.