update keras core.

aimat-lab · Sep 4, 2023 · bfa1edb · bfa1edb
1 parent 2ba17b9
commit bfa1edb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/kgcnn/layers_core/modules.py b/kgcnn/layers_core/modules.py
@@ -34,8 +34,12 @@ def __init__(self,
             trainable=True,
         )
 
+    def build(self, input_shape):
+        super(Embedding, self).build(input_shape)
+
     def call(self, inputs):
         return ops.take(self.embeddings, inputs)
 
     def get_config(self):
         return super(Embedding, self).get_config()
+