From 8ab428fc91eec62006b9df243dd77c7de0111c0f Mon Sep 17 00:00:00 2001 From: AlexanderLavelle Date: Tue, 31 Jan 2023 08:56:29 -0500 Subject: [PATCH] Update embedding_bag.py `self.input_length` is not defined, which affects the pickling / unpickling of this layer in `tf` model mode. --- tensorflow_addons/layers/embedding_bag.py | 1 - 1 file changed, 1 deletion(-) 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()