Merge pull request #3586 from flairNLP/GH-3585-deserialization

GH-3585: Fix DefaultClassifier deserialization
flairNLP · Dec 19, 2024 · 29feea4 · 29feea4
2 parents 016079e + 9359232
commit 29feea4
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: