From 16053c4ae69808c6ce36b10816dc79d66cd19936 Mon Sep 17 00:00:00 2001
From: iperov <lepersorium@gmail.com>
Date: Sun, 21 Apr 2019 22:17:41 +0400
Subject: [PATCH] SAE: revert 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 33915c5..ba755c6 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 = []