Skip to content
Snippets Groups Projects
Commit 6dd47bc4 authored by Yoann Schneider's avatar Yoann Schneider :tennis:
Browse files

Merge branch 'remove-lr-metric' into 'main'

Remove invalid tensorboard logging for lr scheduler metric

Closes #304

See merge request !451
parents f0aad16a b1e123ae
No related branches found
No related tags found
1 merge request!451Remove invalid tensorboard logging for lr scheduler metric
......@@ -695,10 +695,6 @@ class GenericTrainingManager:
self.lr_schedulers[model_name].step(
len(batch_metrics["names"])
)
if "lr" in metric_names:
self.writer.add_scalar(
"lr/{}".format(model_name),
)
# Update dropout scheduler
self.dropout_scheduler.step(len(batch_metrics["names"]))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment