diff options
| author | Ming <a@ufoym.com> | 2017-12-11 14:25:45 +0800 |
|---|---|---|
| committer | Ming <a@ufoym.com> | 2017-12-11 14:25:45 +0800 |
| commit | b455b4c03ef99933543203c373f12d0692c4ae88 (patch) | |
| tree | 5bda539faedfc49e6dfe304420006f5d3c499b9c /models/networks.py | |
| parent | 25da17a66bf0915010c74f40697814502f973579 (diff) | |
Fix `TypeError` when using sigmoid
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)): |