This commit is contained in:
thomashacker 2023-01-19 12:38:28 +01:00
parent 60661ab0fa
commit 97c86072e5

View File

@ -221,12 +221,13 @@ def train_while_improving(
exclude=exclude, exclude=exclude,
annotates=annotating_components, annotates=annotating_components,
) )
nlp.rehearse( if use_rehearse:
subbatch, nlp.rehearse(
losses=losses, subbatch,
sgd=False, # type: ignore[arg-type] losses=losses,
exclude=exclude, sgd=False, # type: ignore[arg-type]
) exclude=exclude,
)
# TODO: refactor this so we don't have to run it separately in here # TODO: refactor this so we don't have to run it separately in here
for name, proc in nlp.pipeline: for name, proc in nlp.pipeline:
if ( if (