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

err1

Browse Source
This commit is contained in:
James Betker 2020-07-10 22:50:56 -06:00
parent ef9f1307eb
commit 716433db1f
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

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