570cd4f66bdf9c3e4b4bb8fbc31fa24b0bd7cdbd,implementations/aae/aae.py,Discriminator,__init__,#Discriminator#,82

Before Change

    def __init__(self):
        super(Discriminator, self).__init__()

        layers = [  nn.Linear(opt.latent_dim, 512),
                    nn.LeakyReLU(0.2, inplace=True),
                    nn.Linear(512, 256),
                    nn.LeakyReLU(0.2, inplace=True),
                    nn.Linear(256, 1),
                    nn.Sigmoid() ]

        self.model = nn.Sequential(*layers)

    def forward(self, latent):
        validity = self.model(latent)

After Change

    def __init__(self):
        super(Discriminator, self).__init__()

        self.model = nn.Sequential(
            nn.Linear(opt.latent_dim, 512),
            nn.LeakyReLU(0.2, inplace=True),
            nn.Linear(512, 256),
            nn.LeakyReLU(0.2, inplace=True),
            nn.Linear(256, 1),
            nn.Sigmoid()
        )

    def forward(self, latent):
        validity = self.model(latent)

Instances