diff --git a/parser/structs/buckets/list_multibucket.py b/parser/structs/buckets/list_multibucket.py index f3387a2..1dab5f9 100644 --- a/parser/structs/buckets/list_multibucket.py +++ b/parser/structs/buckets/list_multibucket.py @@ -56,6 +56,8 @@ def reset(self): self._indices = [[]] self._tokens = [[]] self._str2idx = {'': 0} + for bucket in self: + bucket.reset() return #============================================================= diff --git a/parser/structs/vocabs/multivocabs.py b/parser/structs/vocabs/multivocabs.py index 8154830..fbb984e 100644 --- a/parser/structs/vocabs/multivocabs.py +++ b/parser/structs/vocabs/multivocabs.py @@ -102,6 +102,10 @@ def count(self, train_conllus): status = (vocab._loaded or vocab.count(train_conllus) if hasattr(vocab, 'count') else True) and status return status + def reset(self): + for vocab in self: + vocab.reset() + return #============================================================= def get_input_tensor(self, reuse=True): """"""