It seems this line of code:



 kungkookie:

x = torch.cat((x, low_level_feat), dim=1)


is increasing the number of input channels, so you would have to set in_channels of self.last_conv to 2096 based on the error message.