From 29d1098127c1ce09a989c4058fef609068f33d95 Mon Sep 17 00:00:00 2001 From: Sangwon Lee Date: Wed, 3 Apr 2019 23:11:21 +0900 Subject: [PATCH] fix typo --- src/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/networks.py b/src/networks.py index 3c5b0ee..5e187c0 100644 --- a/src/networks.py +++ b/src/networks.py @@ -504,7 +504,7 @@ def __init__(self, n_in, n_out, kernel_size, stride, padding=0, norm='None', sn= model += [spectral_norm(nn.Conv2d(n_in, n_out, kernel_size=kernel_size, stride=stride, padding=0, bias=True))] else: model += [nn.Conv2d(n_in, n_out, kernel_size=kernel_size, stride=stride, padding=0, bias=True)] - if 'norm' == 'Instance': + if norm == 'Instance': model += [nn.InstanceNorm2d(n_out, affine=False)] model += [nn.LeakyReLU(inplace=True)] self.model = nn.Sequential(*model)