This commit is contained in:
James Betker 2020-07-10 22:52:02 -06:00
parent 716433db1f
commit b3a2c21250

View File

@ -177,7 +177,7 @@ class Discriminator_UNet(nn.Module):
def __init__(self, in_nc, nf): def __init__(self, in_nc, nf):
super(Discriminator_UNet, self).__init__() super(Discriminator_UNet, self).__init__()
# [64, 128, 128] # [64, 128, 128]
self.conv0_0 = ConvGnLelu(in_nc, nf, kernel_size=3, bias=True, gn=False) self.conv0_0 = ConvGnLelu(in_nc, nf, kernel_size=3, bias=True, activation=False)
self.conv0_1 = ConvGnLelu(nf, nf, kernel_size=3, stride=2, bias=False) self.conv0_1 = ConvGnLelu(nf, nf, kernel_size=3, stride=2, bias=False)
# [64, 64, 64] # [64, 64, 64]
self.conv1_0 = ConvGnLelu(nf, nf * 2, kernel_size=3, bias=False) self.conv1_0 = ConvGnLelu(nf, nf * 2, kernel_size=3, bias=False)