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

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(): if self.is_first_run():
conv_weights_list = [] 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: for layer in model.layers:
if type(layer) == keras.layers.Conv2D: if type(layer) == keras.layers.Conv2D:
conv_weights_list += [layer.weights[0]] #Conv2D kernel_weights conv_weights_list += [layer.weights[0]] #Conv2D kernel_weights