From 6a0a27e5c21b90707069184d8e05f39000338750 Mon Sep 17 00:00:00 2001 From: Matthw Honnibal Date: Wed, 1 Jul 2020 18:08:14 +0200 Subject: [PATCH] Fix max_steps --- spacy/cli/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/cli/train.py b/spacy/cli/train.py index 0853bfb56..c053e624c 100644 --- a/spacy/cli/train.py +++ b/spacy/cli/train.py @@ -462,7 +462,7 @@ def train_while_improving( if patience and (step - best_step) >= patience: break # Stop if we've exhausted our max steps (if specified) - if max_steps and (step * accumulate_gradient) >= max_steps: + if max_steps and step >= max_steps: break