Update initialize.py

This commit is contained in:
Ines Montani 2020-09-29 12:14:47 +02:00
parent 42f0e4c946
commit 612bbf85ab

View File

@ -55,7 +55,7 @@ def init_nlp(config: Config, *, use_gpu: int = -1, silent: bool = True) -> Langu
msg.info(f"Resuming training for: {resume_components}")
nlp.resume_training(sgd=optimizer)
with nlp.select_pipes(disable=[*frozen_components, *resume_components]):
nlp.initialize(lambda: train_corpus(nlp), settings=I)
nlp.initialize(lambda: train_corpus(nlp), sgd=optimizer, settings=I)
msg.good("Initialized pipeline components")
# Verify the config after calling 'initialize' to ensure labels
# are properly initialized