Another question. My network takes the input image and outputs an image of same size. Then i add my input image to the ouput image and that should be equal to the ground image. → Input + Net(Input) = Ground Truth. Do i need to add the input image already in the Net class or can i add the input image to the output by torch.add(input,output) just before calculating the loss?