570cd4f66bdf9c3e4b4bb8fbc31fa24b0bd7cdbd,implementations/aae/aae.py,Decoder,__init__,#Decoder#,63

Before Change

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

        layers = [  nn.Linear(opt.latent_dim, 512),
                    nn.LeakyReLU(0.2, inplace=True),
                    nn.Linear(512, 512),
                    nn.BatchNorm1d(512),
                    nn.LeakyReLU(0.2, inplace=True),
                    nn.Linear(512, opt.img_size**2),
                    nn.Tanh()]

        self.model = nn.Sequential(*layers)

    def forward(self, noise):
        img_flat = self.model(noise)
        img = img_flat.view(img_flat.shape[0], opt.channels, opt.img_size, opt.img_size)

After Change

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

        self.model = nn.Sequential(
            nn.Linear(opt.latent_dim, 512),
            nn.LeakyReLU(0.2, inplace=True),
            nn.Linear(512, 512),
            nn.BatchNorm1d(512),
            nn.LeakyReLU(0.2, inplace=True),
            nn.Linear(512, opt.img_size**2),
            nn.Tanh()
        )

    def forward(self, noise):
        img_flat = self.model(noise)
        img = img_flat.view(img_flat.shape[0], opt.channels, opt.img_size, opt.img_size)

Instances