diff options
Diffstat (limited to 'models/networks.py')
| -rwxr-xr-x | models/networks.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/networks.py b/models/networks.py index bf9f14b..a7f9818 100755 --- a/models/networks.py +++ b/models/networks.py @@ -362,7 +362,7 @@ class NLayerDiscriminator(nn.Module): sequence += [[nn.Conv2d(nf, 1, kernel_size=kw, stride=1, padding=padw)]] if use_sigmoid: - sequence += [nn.Sigmoid()] + sequence += [[nn.Sigmoid()]] if getIntermFeat: for n in range(len(sequence)): |