Call resume_training for base model in train CLI

This commit is contained in:
Matthw Honnibal 2019-10-17 21:09:41 +02:00
parent 3f26c50a4d
commit 6aa1c53b1b

View File

@ -234,7 +234,7 @@ def train(
if base_model:
# 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
optimizer = nlp.begin_training(lambda: corpus.train_tuples, device=use_gpu)