From ab1d165bba1ea77c723295cfff3c11772c057218 Mon Sep 17 00:00:00 2001 From: Ines Montani Date: Fri, 14 Aug 2020 14:59:22 +0200 Subject: [PATCH] Pass optimizer defined in config to resume/begin_training Otherwise, this would create a default optimizer, which isn't what we want? --- spacy/cli/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/cli/train.py b/spacy/cli/train.py index fbdb91ab9..c69fb92ea 100644 --- a/spacy/cli/train.py +++ b/spacy/cli/train.py @@ -102,9 +102,9 @@ def train( if resume_components: with nlp.select_pipes(enable=resume_components): msg.info(f"Resuming training for: {resume_components}") - nlp.resume_training() + nlp.resume_training(sgd=optimizer) with nlp.select_pipes(disable=[*frozen_components, *resume_components]): - nlp.begin_training(lambda: train_corpus(nlp)) + nlp.begin_training(lambda: train_corpus(nlp), sgd=optimizer) if tag_map: # Replace tag map with provided mapping