diff --git a/facelib/FANSegmentator.py b/facelib/FANSegmentator.py index 4a8d760..09e14af 100644 --- a/facelib/FANSegmentator.py +++ b/facelib/FANSegmentator.py @@ -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: diff --git a/nnlib/nnlib.py b/nnlib/nnlib.py index 0394910..c52829d 100644 --- a/nnlib/nnlib.py +++ b/nnlib/nnlib.py @@ -938,7 +938,7 @@ class CAInitializerMPSubprocessor(Subprocessor): #override def on_clients_initialized(self): - io.progress_bar ("Processing", len (self.idx_shapes_list)) + io.progress_bar ("Initializing", len (self.idx_shapes_list)) #override def on_clients_finalized(self):