explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-05 04:40:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Pass step=0 to Schedule class to yield initial learning rate

Browse Source
This commit is contained in:
shademe 2023-01-09 18:20:09 +01:00
parent b510fbd0aa
commit da793f9dc1
No known key found for this signature in database
GPG Key ID: 6FCA9FC635B2A402
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/training/loop.py
View File

@ -100,7 +100,7 @@ def train(
stdout.write( stdout.write(
msg.info(f"Set annotations on update for: {annotating_components}") + "\n" msg.info(f"Set annotations on update for: {annotating_components}") + "\n"
) )
stdout.write(msg.info(f"Initial learn rate: {optimizer.learn_rate}") + "\n") stdout.write(msg.info(f"Initial learn rate: {optimizer.learn_rate(step=0)}") + "\n")
with nlp.select_pipes(disable=frozen_components): with nlp.select_pipes(disable=frozen_components):
log_step, finalize_logger = train_logger(nlp, stdout, stderr) log_step, finalize_logger = train_logger(nlp, stdout, stderr)
try: try: