diff --git a/spacy/cli/train.py b/spacy/cli/train.py index 85a29f807..0853bfb56 100644 --- a/spacy/cli/train.py +++ b/spacy/cli/train.py @@ -337,9 +337,9 @@ def create_evaluation_callback(nlp, optimizer, corpus, cfg): if optimizer.averages: with nlp.use_params(optimizer.averages): - scorer = nlp.evaluate(dev_examples, batch_size=eval_batch_size) + scorer = nlp.evaluate(dev_examples, batch_size=batch_size) else: - scorer = nlp.evaluate(dev_examples, batch_size=eval_batch_size) + scorer = nlp.evaluate(dev_examples, batch_size=batch_size) end_time = timer() wps = n_words / (end_time - start_time) scores = scorer.scores