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

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

View File

@ -100,7 +100,7 @@ def train(
stdout.write(
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):
log_step, finalize_logger = train_logger(nlp, stdout, stderr)
try: