github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-14 02:37:00 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

_

Browse source
This commit is contained in:
iperov 2019-03-25 13:28:08 +04:00
commit bb02cc1d97
2 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
facelib/FANSegmentator.py
View file

@ -22,10 +22,9 @@ class FANSegmentator(object):
self.model.load_weights (str(self.weights_path))
else:
if training:
io.log_info ("Initializing CA weights...")
conv_weights_list = []
for layer in self.model.layers:
if type(layer) == Conv2D:
if type(layer) == keras.layers.Conv2D:
conv_weights_list += [layer.weights[0]] # Conv2D kernel_weights
CAInitializerMP(conv_weights_list)
if training: