_

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:

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):