Method forward

cnn-CGANCode/models.py:136–143

(self,input,label)

            torch.nn.Sigmoid()
        )
    def forward(self,input,label):
        x = self.conv_layer_1(input)
        y = self.conv_layer_2(label)
        x = torch.cat([x,y], dim = 1)
        x = self.conv_layer_3(x)
        x = self.conv_layer_4(x)
        out = self.final_layer(x)
        return out