SAE: revert back to batchnormalization

This commit is contained in:
iperov committed 2019-04-21 22:17:41 +04:00
1 parent bef03eec81
commit 16053c4ae6
1 file changed
+1 -1
+1 -1
View File
@@ -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 = []