mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-15 06:09:01 +03:00
Call resume_training for base model in train CLI
This commit is contained in:
parent
3f26c50a4d
commit
6aa1c53b1b
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue
Block a user