From b3a2c21250ffb2d1a777242c76667ecddb83a2f2 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 10 Jul 2020 22:52:02 -0600 Subject: [PATCH] err2 --- codes/models/archs/discriminator_vgg_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/discriminator_vgg_arch.py b/codes/models/archs/discriminator_vgg_arch.py index e0f7ca78..0a109247 100644 --- a/codes/models/archs/discriminator_vgg_arch.py +++ b/codes/models/archs/discriminator_vgg_arch.py @@ -177,7 +177,7 @@ class Discriminator_UNet(nn.Module): def __init__(self, in_nc, nf): super(Discriminator_UNet, self).__init__() # [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) # [64, 64, 64] self.conv1_0 = ConvGnLelu(nf, nf * 2, kernel_size=3, bias=False)