Solved by ptrblck in post #2 
              
              
                Changing the in_channels attribute after creating the module won’t reinitialize the weights and would need to assign a new nn.Conv2d object to self.model.conv1, which accepts 9 input channels.