diff options
| author | Ting-Chun Wang <tcwang0509@berkeley.edu> | 2017-12-11 20:04:34 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-12-11 20:04:34 -0800 |
| commit | e9e6ef6b022954953603ee8d872bce677be7a6c3 (patch) | |
| tree | 5bda539faedfc49e6dfe304420006f5d3c499b9c | |
| parent | 8133fbe49a3b9d3b813c3c2fd8b761708ed02489 (diff) | |
| parent | b455b4c03ef99933543203c373f12d0692c4ae88 (diff) | |
Merge pull request #4 from ufoym/master
Fix `TypeError` when using sigmoid
| -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)): |