From cf2c4fef61f91573ef6117d650db007c1ab1dfe0 Mon Sep 17 00:00:00 2001
From: Timi <timitoc@yahoo.com>
Date: Thu, 3 Oct 2019 22:39:11 +0300
Subject: [PATCH] Fixed subnorm for nonspectral normalization

---
 models/networks/normalization.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/networks/normalization.py b/models/networks/normalization.py
index f4977b66..d5c2a59c 100644
--- a/models/networks/normalization.py
+++ b/models/networks/normalization.py
@@ -26,6 +26,8 @@ def add_norm_layer(layer):
         if norm_type.startswith('spectral'):
             layer = spectral_norm(layer)
             subnorm_type = norm_type[len('spectral'):]
+        else:
+            subnorm_type = norm_type
 
         if subnorm_type == 'none' or len(subnorm_type) == 0:
             return layer