diff --git a/stanza/models/coref/model.py b/stanza/models/coref/model.py index 59721dab41..70a4460f6a 100644 --- a/stanza/models/coref/model.py +++ b/stanza/models/coref/model.py @@ -198,7 +198,7 @@ def load_weights(self, elif key.endswith("_scheduler"): self.schedulers[key].load_state_dict(state_dict) else: - self.trainable[key].load_state_dict(state_dict) + self.trainable[key].load_state_dict(state_dict, strict=False) print(f"Loaded {key}") def run(self, # pylint: disable=too-many-locals