fixed a bug: the stride of the last two discriminator layers should be 1

Author: taesungp

models/networks/discriminator.py

@@ -87,8 +87,9 @@ def __init__(self, opt):
  for n in range(1, opt.n_layers_D):
  nf_prev = nf
  nf = min(nf * 2, 512)
+ stride = 1 if n == opt.n_layers_D - 1 else 2
  sequence += [[norm_layer(nn.Conv2d(nf_prev, nf, kernel_size=kw,
- stride=2, padding=padw)),
+ stride=stride, padding=padw)),
  nn.LeakyReLU(0.2, False)
  ]]