e4dedd15f0dd909f48385867b874519e57ea0363,implementations/aae/aae.py,Encoder,__init__,#Encoder#,45

Before Change

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

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

        self.model = nn.Sequential(*layers)

    def forward(self, img):
        img_flat = img.view(img.shape[0], -1)
        latent = self.model(img_flat)

After Change

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

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

    def forward(self, img):
        img_flat = img.view(img.shape[0], -1)
        latent = self.model(img_flat)

Instances