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

_

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
facelib/FANSegmentator.py
+1 -2
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:
nnlib/nnlib.py
+1 -1
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):