From b455b4c03ef99933543203c373f12d0692c4ae88 Mon Sep 17 00:00:00 2001 From: Ming Date: Mon, 11 Dec 2017 14:25:45 +0800 Subject: Fix `TypeError` when using sigmoid --- models/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'models/networks.py') 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)): -- cgit v1.2.3-70-g09d2