models.py [140:151]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self.decoder = nn.Sequential(
            nn.Linear(z_dim, 256),
            nn.ReLU(),
            Lambda(lambda x: x.view(-1, 256, 1, 1)),
            nn.ConvTranspose2d(256, 64, 4),
            nn.ReLU(),
            nn.ConvTranspose2d(64, 64, 4, 2, 1),
            nn.ReLU(),
            nn.ConvTranspose2d(64, n_pixels, 4, 2, 1),
            nn.ReLU(),
            nn.ConvTranspose2d(n_pixels, n_channels, 4, 2, 1),
            Lambda(lambda x: x.view(x.size(0), -1)),
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



models.py [274:285]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self.decoder = nn.Sequential(
            nn.Linear(z_dim, 256),
            nn.ReLU(),
            Lambda(lambda x: x.view(-1, 256, 1, 1)),
            nn.ConvTranspose2d(256, 64, 4),
            nn.ReLU(),
            nn.ConvTranspose2d(64, 64, 4, 2, 1),
            nn.ReLU(),
            nn.ConvTranspose2d(64, n_pixels, 4, 2, 1),
            nn.ReLU(),
            nn.ConvTranspose2d(n_pixels, n_channels, 4, 2, 1),
            Lambda(lambda x: x.view(x.size(0), -1)),
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -