Update embedding_bag.py (#2804)

`self.input_length` is not defined, which affects the pickling / unpickling of this layer in `tf` model mode.
tensorflow · Feb 23, 2023 · 104e345 · 104e345
1 parent 14af819
commit 104e345
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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()