update to Keras 2.2.2

2018-11-14 07:22:27 -08:00
parent c90d67bec1
commit 57fd6735a7
3 changed files with 4 additions and 4 deletions
@@ -57,7 +57,7 @@ class LandmarksExtractor(object):
    def __init__ (self, keras):
        self.keras = keras
        K = self.keras.backend
-        class TorchBatchNorm2D(self.keras.engine.topology.Layer):
+        class TorchBatchNorm2D(self.keras.layers.Layer):
            def __init__(self, axis=-1, momentum=0.99, epsilon=1e-3, **kwargs):
                super(TorchBatchNorm2D, self).__init__(**kwargs)
                self.supports_masking = True