Training on other datasets?

[marvision-ai]

Great Tutorial Tony! Thank you for writting it out.

Do you think you could potentially go over training the UNet with your own dataset?

That would be very valuable and interesting.

Thanks!