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

fix AVATARModel

Browse Source
This commit is contained in:
iperov
2019-08-24 19:55:38 +04:00
parent 2a9a579436
commit 5968ac21f6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/Model_AVATAR/Model.py
View File

@@ -67,7 +67,7 @@ class AVATARModel(ModelBase):
if self.is_first_run():
conv_weights_list = []
for model in [self.enc, self.decA64, self.decB64, self.C, self.D, self.CD]:
for model in self.get_model_filename_list():
for layer in model.layers:
if type(layer) == keras.layers.Conv2D:
conv_weights_list += [layer.weights[0]] #Conv2D kernel_weights