SAE: back to batchnormalization

2019-04-22 10:28:01 +04:00
parent 3aef9a507b
commit cd2767cca0
1 changed files with 1 additions and 1 deletions
@@ -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 = []