Pass optimizer to TrainablePipe.distill

This commit is contained in:
Daniël de Kok 2023-01-20 14:57:19 +01:00
parent 8ca1cf7355
commit 3ce9a3d2c4

View File

@ -1092,12 +1092,10 @@ class Language:
student_proc.distill( student_proc.distill(
teacher_pipe, teacher_pipe,
examples, examples,
sgd=None, sgd=sgd,
losses=losses, losses=losses,
**component_cfg[name], **component_cfg[name],
) )
if sgd is not None:
student_proc.finish_update(sgd)
if name in annotates: if name in annotates:
for doc, eg in zip( for doc, eg in zip(
_pipe( _pipe(