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: back to batchnormalization

Browse source
This commit is contained in:
iperov 2019-04-22 10:28:01 +04:00
parent 3aef9a507b
commit cd2767cca0
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 = []