Sir, I normalised the images and masks to have mean 0.5 and std 0.5. I also rechecked and the min and max values of images and masks are:
image -->max and min  tensor(1.) tensor(-0.9294)
mask--> max ad min tensor(1.) tensor(-1.)


What should I do ?