when output channel number C‘ are not equal to input channel number C,there is different channel numbers between identity with self.k2(x). So how to deal with this problem? The code as follows:
out = torch.sigmoid(torch.add(identity, F.interpolate(self.k2(x), identity.size()[2:])))
when output channel number C‘ are not equal to input channel number C,there is different channel numbers between identity with self.k2(x). So how to deal with this problem? The code as follows:
out = torch.sigmoid(torch.add(identity, F.interpolate(self.k2(x), identity.size()[2:])))