forked from mrq/DL-Art-School
err2
This commit is contained in:
parent
716433db1f
commit
b3a2c21250
|
@ -177,7 +177,7 @@ class Discriminator_UNet(nn.Module):
|
|||
def __init__(self, in_nc, nf):
|
||||
super(Discriminator_UNet, self).__init__()
|
||||
# [64, 128, 128]
|
||||
self.conv0_0 = ConvGnLelu(in_nc, nf, kernel_size=3, bias=True, gn=False)
|
||||
self.conv0_0 = ConvGnLelu(in_nc, nf, kernel_size=3, bias=True, activation=False)
|
||||
self.conv0_1 = ConvGnLelu(nf, nf, kernel_size=3, stride=2, bias=False)
|
||||
# [64, 64, 64]
|
||||
self.conv1_0 = ConvGnLelu(nf, nf * 2, kernel_size=3, bias=False)
|
||||
|
|
Loading…
Reference in New Issue
Block a user