Method forward

Code/net/Discriminator.py:53–56

(self,x)

            *layers
        )
    def forward(self,x):
        x = self.initial(x)
        out= self.model(x)
        return out

if __name__ == '__main__':
    x = torch.randn(size = (5,3,256,256),device='cpu')