Solved by ptrblck in post #4 
              
              
                You haven’t posted the code of your model, but I assume the last layer is a convolution with out_channels=3. 
You can add code using three backticks `. It’ll make debugging easier and the search of this forum can find your code, if someone else has this problem. 
Are you dealing with gray-scale imag…