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

Before Change

class UNetDown(nn.Module):
    def __init__(self, in_size, out_size, bn=True, dropout=0.0):
        super(UNetDown, self).__init__()
        model = [   nn.Conv2d(in_size, out_size, 3, stride=2, padding=1),
                    nn.LeakyReLU(0.2, inplace=True) ]

        if bn:
            model += [nn.InstanceNorm2d(out_size)]

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

        self.model = nn.Sequential(*model)

After Change

                        nn.InstanceNorm2d(in_features),
                        nn.ReLU(inplace=True),
                        nn.ReflectionPad2d(1),
                        nn.Conv2d(in_features, in_features, 3),
                        nn.InstanceNorm2d(in_features)  ]

        self.conv_block = nn.Sequential(*conv_block)

Instances