From 78b952b5f07426363cb61d7309ba1a516c8d431f Mon Sep 17 00:00:00 2001 From: Benedikt Fuchs Date: Mon, 6 Mar 2023 17:44:27 +0100 Subject: [PATCH] make transformer training vocab optional --- flair/trainers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flair/trainers/trainer.py b/flair/trainers/trainer.py index bfbde8ab2..7243584d7 100644 --- a/flair/trainers/trainer.py +++ b/flair/trainers/trainer.py @@ -294,7 +294,7 @@ def train( weight_extractor = WeightExtractor(base_path) with contextlib.ExitStack() as context_stack: - if isinstance(self.model, ReduceTransformerVocabMixin): + if reduce_transformer_vocab and isinstance(self.model, ReduceTransformerVocabMixin): transformer_embeddings = get_transformer_embeddings(self) if not transformer_embeddings: reduce_transformer_vocab = False