61228f3243eaee39cba31a9abd2afc6ead9612a8,implementations/pix2pix/models.py,UNetDown,__init__,#UNetDown#,10

Before Change

                    nn.LeakyReLU(0.2, inplace=True) ]

        if normalize:
            model += [nn.BatchNorm2d(out_size, 0.8)]

        if dropout:
            model += [nn.Dropout(dropout)]

        self.model = nn.Sequential(*model)

    def forward(self, x):
        return self.model(x)

After Change

            layers.append(nn.InstanceNorm2d(out_size))
        layers.append(nn.LeakyReLU(0.2, inplace=True))
        if dropout:
            layers.append(nn.Dropout(dropout))
        self.model = nn.Sequential(*layers)

    def forward(self, x):
        return self.model(x)

Instances