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

fix dimensions error

Browse source
This commit is contained in:
iperov 2020-12-11 14:34:37 +04:00
parent ca03bbca04
commit 35945b257c
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/leras/archis/DeepFakeArchi.py
View file

@ -87,7 +87,7 @@ class DeepFakeArchi(nn.ArchiBase):
return res // (2**4) return res // (2**4)
def get_out_ch(self): def get_out_ch(self):
return 512 return self.e_ch * 8
lowest_dense_res = resolution // (32 if 'd' in opts else 16) lowest_dense_res = resolution // (32 if 'd' in opts else 16)

2
models/Model_SAEHD/Model.py
View file

@ -248,7 +248,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
if self.is_training: if self.is_training:
if self.options['true_face_power'] != 0: if self.options['true_face_power'] != 0:
self.code_discriminator = nn.CodeDiscriminator(ae_dims, code_res=model_archi.Inter.get_out_res(), name='dis' ) self.code_discriminator = nn.CodeDiscriminator(ae_dims, code_res=self.inter.get_out_res(), name='dis' )
self.model_filename_list += [ [self.code_discriminator, 'code_discriminator.npy'] ] self.model_filename_list += [ [self.code_discriminator, 'code_discriminator.npy'] ]
elif 'liae' in archi_type: elif 'liae' in archi_type: