SAE: revert back to batchnormalization

2019-04-21 22:17:41 +04:00
parent bef03eec81
commit 16053c4ae6
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 = []