diff options
Diffstat (limited to 'models')
| -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 585b940..f48e7fe 100644 --- a/models/networks.py +++ b/models/networks.py @@ -26,7 +26,7 @@ def get_norm_layer(norm_type='instance'): elif norm_type == 'instance': norm_layer = functools.partial(nn.InstanceNorm2d, affine=False) else: - raise NotImplementedError('normalization layer [%s] is not found' % norm) + raise NotImplementedError('normalization layer [%s] is not found' % norm_type) return norm_layer |