3b13f1e6b5f5ac19511cb80f8107d4a1e077beca,model.py,Discriminator,forward,#Discriminator#Any#,84

Before Change

    def forward(self, x):
        batch_size = x.size(0)
        return self.net(x).view(batch_size)


class ResidualBlock(nn.Module):
    def __init__(self, channels):

After Change

    def forward(self, x):
        batch_size = x.size(0)
        x = self.net(x)
        x = x.view(batch_size, -1)
        return F.sigmoid(self.classifier(x))


class ResidualBlock(nn.Module):
    def __init__(self, channels):

Instances