Fix type annotation, default config value

This commit is contained in:
shademe 2022-11-02 18:39:52 +01:00
parent f64dbe6925
commit 43e5fb7b6b
No known key found for this signature in database
GPG Key ID: 6FCA9FC635B2A402
2 changed files with 3 additions and 1 deletions

View File

@ -90,6 +90,8 @@ dev_corpus = "corpora.dev"
train_corpus = "corpora.train"
# Optional callback before nlp object is saved to disk after training
before_to_disk = null
# Optional callback that is invoked at the start of each training step
before_update = null
[training.logger]
@loggers = "spacy.ConsoleLogger.v1"

View File

@ -152,7 +152,7 @@ def train_while_improving(
max_steps: int,
exclude: List[str],
annotating_components: List[str],
before_update: Optional[Callable[[Language, int], None]],
before_update: Optional[Callable[["Language", int], None]],
):
"""Train until an evaluation stops improving. Works as a generator,
with each iteration yielding a tuple `(batch, info, is_best_checkpoint)`,