Fix vgg disc arch

2020-05-27 13:31:22 -06:00 · 2020-05-27 13:31:22 -06:00 · f745be9dea
commit f745be9dea
parent 6962ccb306
1 changed files with 1 additions and 1 deletions
--- 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)))