Oops

stanfordnlp · Dec 24, 2024 · 667e35c · 667e35c
1 parent 9578401
commit 667e35c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stanza/models/lemma/trainer.py b/stanza/models/lemma/trainer.py
@@ -32,10 +32,10 @@ def unpack_batch(batch, device):
 
 class Trainer(object):
     """ A trainer for training models. """
-    def __init__(self, args=None, vocab=None, emb_matrix=None, model_file=None, device=None, foundation_cache=None):
+    def __init__(self, args=None, vocab=None, emb_matrix=None, model_file=None, device=None, foundation_cache=None, lemma_classifier_args=None):
         if model_file is not None:
             # load everything from file
-            self.load(model_file, args, foundation_cache)
+            self.load(model_file, args, foundation_cache, lemma_classifier_args)
         else:
             # build model from scratch
             self.args = args