GH-3585: Fix DefaultClassifier deserialization

flairNLP · Dec 19, 2024 · 9359232 · 9359232
1 parent ca959f1
commit 9359232
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flair/nn/model.py b/flair/nn/model.py
@@ -993,7 +993,7 @@ def _get_state_dict(self):
         state["locked_dropout"] = self.locked_dropout.dropout_rate
         state["multi_label"] = self.multi_label
         state["multi_label_threshold"] = self.multi_label_threshold
-        state["loss_weights"] = self.loss_weights
+        state["loss_weights"] = self.weight_dict
         state["train_on_gold_pairs_only"] = self.train_on_gold_pairs_only
         state["inverse_model"] = self.inverse_model
         if self._custom_decoder: