diff --git a/spacy/default_config_distillation.cfg b/spacy/default_config_distillation.cfg index 2993f8e48..1926fafa9 100644 --- a/spacy/default_config_distillation.cfg +++ b/spacy/default_config_distillation.cfg @@ -6,7 +6,7 @@ corpus = "corpora.distillation" dropout = 0.1 max_epochs = 1 max_steps = 0 -pipe_map = {} +student_to_teacher = {} [distillation.batcher] @batchers = "spacy.batch_by_words.v1" diff --git a/spacy/schemas.py b/spacy/schemas.py index 51b64580d..e830eb01f 100644 --- a/spacy/schemas.py +++ b/spacy/schemas.py @@ -418,7 +418,7 @@ class ConfigSchemaDistill(BaseModel): max_epochs: StrictInt = Field(..., title="Maximum number of epochs to distill for") max_steps: StrictInt = Field(..., title="Maximum number of steps to distill for") optimizer: Optimizer = Field(..., title="The optimizer to use") - pipe_map: Dict[str, str] = Field(..., title="Mapping from teacher to student pipe") + student_to_teacher: Dict[str, str] = Field(..., title="Mapping from student to teacher pipe") # fmt: on class Config: