flairNLPGH-3585: Fix DefaultClassifier deserialization

ZipRecruiter · Jan 16, 2025 · 5e9b90b · 5e9b90b
1 parent 2956e02
commit 5e9b90b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flair/nn/model.py b/flair/nn/model.py
@@ -996,7 +996,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: