54d89c639d8d6d72e5ada833c00762b82ba63348,implementations/bicyclegan/models.py,Discriminator,__init__,#Discriminator#Any#,131

Before Change

        self.d1 = nn.Sequential(
            *downsample_block(in_channels, 64, False),
            *downsample_block(64, 128, True),
            nn.Conv2d(128, 1, 3, 1, 1)
        )
        self.d2 = nn.Sequential(
            *downsample_block(in_channels, 64, False),
            *downsample_block(64, 128, True),

After Change

        // Extracts three discriminator models
        self.models = nn.ModuleList()
        for i in range(3):
            self.models.add_module("disc_%d" % i,
                nn.Sequential(
                    *discriminator_block(in_channels, 64, normalize=False),
                    *discriminator_block(64, 128),
                    *discriminator_block(128, 256),
                    *discriminator_block(256, 512),
                    nn.Conv2d(512, 1, 3, padding=1)
                )
            )

        self.downsample = nn.AvgPool2d(in_channels, stride=2, padding=[1, 1], count_include_pad=False)

    def compute_loss(self, x, gt):

Instances