mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

More pixdisc fixes

Browse Source
This commit is contained in:
James Betker 2020-07-05 22:03:16 -06:00
parent 909007ee6a
commit 72f90cabf8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/archs/discriminator_vgg_arch.py
View File

@ -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)