Need default, empty optimizers & schedulers to load & resave a coref …

…model
stanfordnlp · Oct 31, 2023 · 98ea5bf · 98ea5bf
1 parent edf24d1
commit 98ea5bf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stanza/models/coref/model.py b/stanza/models/coref/model.py
@@ -66,6 +66,10 @@ def __init__(self,
         self.epochs_trained = epochs_trained
         self._docs: Dict[str, List[Doc]] = {}
         self._build_model()
+
+        self.optimizers = {}
+        self.schedulers = {}
+
         if build_optimizers:
             self._build_optimizers()
         self._set_training(False)