CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
core/leras/archis/DeepFakeArchi.py
+1 -1
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)
models/Model_SAEHD/Model.py
+1 -1
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: