@mohitsharma916 Thank you. I have done it before. The result is still bad. Do you think we miss something?, because I add log_softmax after nllloss2d. Do you think it necessary?. My groundtruth is in range 0-1(binary classification) and my image is 0-255.