github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

SAE: revert back to batchnormalization

Browse source
This commit is contained in:
iperov 2019-04-21 22:17:41 +04:00
parent bef03eec81
commit 16053c4ae6
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_SAE/Model.py
View file

@ -127,7 +127,7 @@ class SAEModel(ModelBase):
padding = 'reflect' if self.options['remove_gray_border'] else 'zero' padding = 'reflect' if self.options['remove_gray_border'] else 'zero'
common_flow_kwargs = { 'padding': padding, common_flow_kwargs = { 'padding': padding,
'norm': '', 'norm': 'bn',
'act':'' } 'act':'' }
weights_to_load = [] weights_to_load = []