@ptrblck Thank you for spending time and explain to me about issues in my code. Does your unet model works for images with 1 channel? I think your dataset was RGB when you have written this model. what is in_channels_skip in your model?