_

2019-03-25 13:28:08 +04:00
parent aa58d9e563
commit bb02cc1d97
2 changed files with 2 additions and 3 deletions
@@ -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: