github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-08 05:51:40 -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
parent aa58d9e563
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)) self.model.load_weights (str(self.weights_path))
else: else:
if training: if training:
io.log_info ("Initializing CA weights...")
conv_weights_list = [] conv_weights_list = []
for layer in self.model.layers: 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 conv_weights_list += [layer.weights[0]] # Conv2D kernel_weights
CAInitializerMP(conv_weights_list) CAInitializerMP(conv_weights_list)
if training: if training:

2
nnlib/nnlib.py
View file

@ -938,7 +938,7 @@ class CAInitializerMPSubprocessor(Subprocessor):
#override #override
def on_clients_initialized(self): def on_clients_initialized(self):
io.progress_bar ("Processing", len (self.idx_shapes_list)) io.progress_bar ("Initializing", len (self.idx_shapes_list))
#override #override
def on_clients_finalized(self): def on_clients_finalized(self):