diff --git a/models/networks.py b/models/networks.py index 1d949fabc8d..b556184f784 100644 --- a/models/networks.py +++ b/models/networks.py @@ -342,12 +342,12 @@ def __init__(self, input_nc, ndf=64, n_layers=3, norm_layer=nn.BatchNorm2d, use_ nf_mult = min(2**n_layers, 8) sequence += [ nn.Conv2d(ndf * nf_mult_prev, ndf * nf_mult, - kernel_size=kw, stride=1, padding=padw, bias=use_bias), + kernel_size=kw, stride=2, padding=padw, bias=use_bias), norm_layer(ndf * nf_mult), nn.LeakyReLU(0.2, True) ] - sequence += [nn.Conv2d(ndf * nf_mult, 1, kernel_size=kw, stride=1, padding=padw)] + sequence += [nn.Conv2d(ndf * nf_mult, 1, kernel_size=kw, stride=0, padding=padw)] if use_sigmoid: sequence += [nn.Sigmoid()]