Thanks for your answer. So as input i have a numpy array that i change to a pytorch tensor. My ground truth is also a numpy arra that is transformed to a tensor. the values in the numpy array represent the grayscale values of an infrared image. I tried it for only 1 picture and with different learning rates. But i just creates weird output and doesnt seem like its learning. I feel like the update of the parameters is pretty random.