Hi beaupreda,
Thanks and appreciate the information.
I didn’t aware of the inherit feature until you mention it, this sure looks useful.
Actually what i’m trying to do is perform segmentation classification with unet, so in each epoch, i would need to feed in a mask and an image. Do you aware of a better way to do this?
Anyway, i will proceed with your suggestion and see how it would work, thanks again