Rename pipe_map to student_to_teacher

This commit is contained in:
Daniël de Kok 2023-01-31 12:24:38 +01:00
parent b450d34dca
commit 4a7a60c9e6
2 changed files with 2 additions and 2 deletions

View File

@ -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"

View File

@ -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: