Pass optimizer to TrainablePipe.distill

2025-08-02 11:20:19 +03:00 · 2023-01-20 14:57:19 +01:00 · 2023-01-20 14:57:19 +01:00 · 3ce9a3d2c4
commit 3ce9a3d2c4
parent 8ca1cf7355
1 changed files with 1 additions and 3 deletions
--- a/spacy/language.py
+++ b/spacy/language.py
@ -1092,12 +1092,10 @@ class Language:
                student_proc.distill(
                    teacher_pipe,
                    examples,
-                    sgd=None,
+                    sgd=sgd,
                    losses=losses,
                    **component_cfg[name],
                )
-                if sgd is not None:
-                    student_proc.finish_update(sgd)
            if name in annotates:
                for doc, eg in zip(
                    _pipe(