From 72f90cabf83348637018b8ab3a1352e30fb0778a Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 5 Jul 2020 22:03:16 -0600 Subject: [PATCH] More pixdisc fixes --- 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 f99edfc5..02fd156e 100644 --- a/codes/models/archs/discriminator_vgg_arch.py +++ b/codes/models/archs/discriminator_vgg_arch.py @@ -107,7 +107,7 @@ class Discriminator_VGG_PixLoss(nn.Module): self.bn4_1 = nn.BatchNorm2d(nf * 8, affine=True) self.reduce_1 = ConvBnLelu(nf * 8, nf * 4, bias=False) - self.pix_loss_collapse = ConvBnLelu(nf * 4, 1, bias=False) + self.pix_loss_collapse = ConvBnLelu(nf * 4, 1, bias=False, bn=False, lelu=False) # activation function self.lrelu = nn.LeakyReLU(negative_slope=0.2, inplace=True)