From cd2767cca04dce8d8f9f4f7525089c932b5778e5 Mon Sep 17 00:00:00 2001
From: iperov <lepersorium@gmail.com>
Date: Mon, 22 Apr 2019 10:28:01 +0400
Subject: [PATCH] SAE: back to batchnormalization

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

diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py
index be388d1..7147b6f 100644
--- a/models/Model_SAE/Model.py
+++ b/models/Model_SAE/Model.py
@@ -127,7 +127,7 @@ class SAEModel(ModelBase):
 
         padding = 'reflect' if self.options['remove_gray_border'] else 'zero'
         common_flow_kwargs = { 'padding': padding,
-                               'norm': '',
+                               'norm': 'bn',
                                'act':'' }
 
         weights_to_load = []