From 716433db1fb5afe1e14ddf0b22325c21aeb573d0 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 10 Jul 2020 22:50:56 -0600 Subject: [PATCH] err1 --- 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 3f110edb..e0f7ca78 100644 --- a/codes/models/archs/discriminator_vgg_arch.py +++ b/codes/models/archs/discriminator_vgg_arch.py @@ -175,7 +175,7 @@ class Discriminator_VGG_PixLoss(nn.Module): class Discriminator_UNet(nn.Module): def __init__(self, in_nc, nf): - super(Discriminator_VGG_PixLoss, self).__init__() + super(Discriminator_UNet, self).__init__() # [64, 128, 128] self.conv0_0 = ConvGnLelu(in_nc, nf, kernel_size=3, bias=True, gn=False) self.conv0_1 = ConvGnLelu(nf, nf, kernel_size=3, stride=2, bias=False)