diff options
| author | guopzhao <guopzhao@qq.com> | 2017-12-15 14:47:05 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-12-15 14:47:05 +0800 |
| commit | 765a32fc83ac8c477dc5a42f92128bd6de72e4b1 (patch) | |
| tree | be7978cf11a4de7c481c4462787ce9484375625b | |
| parent | 73132d2ded0714946a9068b5cf818da2a43ac6d2 (diff) | |
Update networks.py
| -rw-r--r-- | models/networks.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |