Call resume_training for base model in train CLI

This commit is contained in:
Matthw Honnibal 2019-10-17 21:09:41 +02:00 committed by Matthew Honnibal
parent 3edb297649
commit 4fb31adde5

View File

@ -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}