From 765a32fc83ac8c477dc5a42f92128bd6de72e4b1 Mon Sep 17 00:00:00 2001 From: guopzhao Date: Fri, 15 Dec 2017 14:47:05 +0800 Subject: Update networks.py --- 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 568f8c9..da2f59c 100644 --- a/models/networks.py +++ b/models/networks.py @@ -77,7 +77,7 @@ def get_norm_layer(norm_type='instance'): norm_layer = functools.partial(nn.BatchNorm2d, affine=True) elif norm_type == 'instance': norm_layer = functools.partial(nn.InstanceNorm2d, affine=False) - elif layer_type == 'none': + elif norm_type == 'none': norm_layer = None else: raise NotImplementedError('normalization layer [%s] is not found' % norm_type) -- cgit v1.2.3-70-g09d2