Skip to content
Snippets Groups Projects
Commit b1e123ae authored by Manon Blanco's avatar Manon Blanco
Browse files

Remove invalid tensorboard logging for lr scheduler metric

parent f0aad16a
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