Simon,
Thanks a lot!
I just remove the channel dimension from my masks and everything works well… now I’m generating masks with the shape 3 [width=388, height=388].
After that, I’m working with input images (X), target masks (y) and predicted output masks (y_hat) as follow:
X     --> torch.Size([10, 1, 572, 572])
y     --> torch.Size([10, 388, 388])
y_hat --> torch.Size([10, 2, 388, 388])

But, I don’t understand why target masks (y) and predicted masks (y_hat) must have different shapes? It’s so weird for me…
Here 7 are my jupyter notebook that I’m using in my tests.