From 4fb31adde555617965b498360fb63e9e705196ad Mon Sep 17 00:00:00 2001 From: Matthw Honnibal Date: Thu, 17 Oct 2019 21:09:41 +0200 Subject: [PATCH] Call resume_training for base model in train CLI --- 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 d4de9aeb4..fc4c9f67b 100644 --- a/spacy/cli/train.py +++ b/spacy/cli/train.py @@ -285,7 +285,7 @@ def train( if base_model and not pipes_added: # Start with an existing model, use default optimizer - optimizer = create_default_optimizer(Model.ops) + optimizer = nlp.resume_training(device=use_gpu) else: # Start with a blank model, call begin_training cfg = {"device": use_gpu}