c7f7d1979a35b443dba7e776203ed7084efecf77,models/networks.py,ResnetBlock,build_conv_block,#ResnetBlock#,185

Before Change

        conv_block = []
        p = 0
        // TODO: support padding types
        assert(padding_type == "zero")
        p = 1

        // TODO: InstanceNorm
        conv_block += [nn.Conv2d(dim, dim, kernel_size=3, padding=p),

After Change

            conv_block += [nn.Dropout(0.5)]

        p = 0
        if padding_type == "reflect":
            conv_block += [nn.ReflectionPad2d(1)]
        elif padding_type == "replicate":
            conv_block += [nn.ReplicationPad2d(1)]
        elif padding_type == "zero":
            p = 1
        else:
            raise NotImplementedError("padding [%s] is not implemented" % padding_type)
        conv_block += [nn.Conv2d(dim, dim, kernel_size=3, padding=p),
                       norm_layer(dim)]

        return nn.Sequential(*conv_block)

Instances