61228f3243eaee39cba31a9abd2afc6ead9612a8,implementations/pix2pix/models.py,GeneratorUNet,__init__,#GeneratorUNet#,45

Before Change

        self.up5 = UNetUp(256, 64)


        final = [   nn.Upsample(scale_factor=2),
                    nn.Conv2d(128, out_channels, 3, 1, 1),
                    nn.Tanh() ]
        self.final = nn.Sequential(*final)

    def forward(self, x):
        // U-Net generator with skip connections from encoder to decoder
        d1 = self.down1(x)

After Change

        self.up7 = UNetUp(256, 64)


        self.final = nn.Sequential(
            nn.Upsample(scale_factor=2),
            nn.ZeroPad2d((1, 0, 1, 0)),
            nn.Conv2d(128, out_channels, 4, padding=1),
            nn.Tanh()
        )


    def forward(self, x):
        // U-Net generator with skip connections from encoder to decoder

Instances