Resolve "Flair Embeddings"

Closes #13

poldeepner2/model/xlmr_for_token_classification.py

@@ -15,8 +15,8 @@ class XLMRForTokenClassification(nn.Module):
         self.flair_embeddings = flair_embeddings
         self.n_labels = n_labels
         if flair_embeddings:
-            self.linear_1 = nn.Linear(hidden_size + flair_embed_size, hidden_size + flair_embed_size)
-            self.classification_head = nn.Linear(hidden_size + flair_embed_size, n_labels)
+            self.linear_1 = nn.Linear(hidden_size + flair_embed_size, hidden_size)
+            self.classification_head = nn.Linear(hidden_size, n_labels)
         else:
             self.linear_1 = nn.Linear(hidden_size, hidden_size)
             self.classification_head = nn.Linear(hidden_size, n_labels)