From 75ec31182ca8513633ef964165777468477a449f Mon Sep 17 00:00:00 2001
From: Hungryof <ziquanlanyue@163.com>
Date: Mon, 4 Sep 2017 18:29:38 +0800
Subject: Update networks.py

---
 models/networks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'models')

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
 
 
-- 
cgit v1.2.3-70-g09d2