Fix descripion of max_epochs

This commit is contained in:
Daniël de Kok 2023-01-30 15:50:44 +01:00
parent 6e20b506a0
commit 4b311bcae7

View File

@ -415,7 +415,7 @@ class ConfigSchemaDistill(BaseModel):
batcher: Batcher = Field(..., title="Batcher for the training data")
corpus: StrictStr = Field(..., title="Path in the config to the distillation data")
dropout: StrictFloat = Field(..., title="Dropout rate")
max_epochs: StrictInt = Field(..., title="Maximum number of steps to distill for")
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")