From f745be9deadd71abb1d005c8fcbbfb3d8187046e Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 27 May 2020 13:31:22 -0600 Subject: [PATCH] Fix vgg disc arch --- 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 34ed5f08..29dda1dc 100644 --- a/codes/models/archs/discriminator_vgg_arch.py +++ b/codes/models/archs/discriminator_vgg_arch.py @@ -39,7 +39,7 @@ class Discriminator_VGG_128(nn.Module): self.lrelu = nn.LeakyReLU(negative_slope=0.2, inplace=True) def forward(self, x): - x, skip_med, skip_lo = x + x = x[0] fea = self.lrelu(self.conv0_0(x)) fea = self.lrelu(self.bn0_1(self.conv0_1(fea)))