diff --git a/tensorflow_addons/layers/embedding_bag.py b/tensorflow_addons/layers/embedding_bag.py index 046b51207a..de415d9081 100644 --- a/tensorflow_addons/layers/embedding_bag.py +++ b/tensorflow_addons/layers/embedding_bag.py @@ -147,7 +147,6 @@ def get_config(self): self.embeddings_constraint ), "mask_zero": self.mask_zero, - "input_length": self.input_length, "combiner": self.combiner, } base_config = super(EmbeddingBag, self).get_config()