Merge pull request #2073 from djstrong/patch-8

Fix linear input dimension if the reproject
flairNLP · Feb 2, 2021 · 69a4921 · 69a4921
2 parents 5473bfa + e6f2c9d
commit 69a4921
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flair/models/sequence_tagger_model.py b/flair/models/sequence_tagger_model.py
@@ -213,7 +213,7 @@ def __init__(
             )
         else:
             self.linear = torch.nn.Linear(
-                self.embeddings.embedding_length, len(tag_dictionary)
+                rnn_input_dim, len(tag_dictionary)
             )
 
         if self.use_crf: