Solved by ptrblck in post #29 
              
              
                Thanks for the code. 
I’m not sure, if you would need this workaround and why you are replacing the self.model.conv1 with a conv layer accepting a single input channel. 
The default resnet18 model already accepts RGB images, so you could just remove the self.model.conv1 line of code as well as self.…