Hello. I have a Unet network for segmentation. Using dice loss  the net is learning. But when I use Crossentropy, after one epoch the loss is constant. I ve searched on internet and i ve found a function to check gradients, and after 3 epochs they become 0. Can someone help me with some ideas?
I already use Relu, and init the conv2d with xavier.