Unverified Commit 8940d315 authored by Claudius Kienle's avatar Claudius Kienle Committed by GitHub
Browse files

Trainer: fixed evaluate raising `KeyError` for ReduceLROnPlateau (#23952)



Trainer: fixed KeyError on evaluate for ReduceLROnPlateau
Co-authored-by: default avatarClaudius Kienle <claudius.kienle@artiminds.com>
parent 2fdba73a
......@@ -2320,7 +2320,10 @@ class Trainer:
# Run delayed LR scheduler now that metrics are populated
if isinstance(self.lr_scheduler, torch.optim.lr_scheduler.ReduceLROnPlateau):
self.lr_scheduler.step(metrics[self.args.metric_for_best_model])
metric_to_check = self.args.metric_for_best_model
if not metric_to_check.startswith("eval_"):
metric_to_check = f"eval_{metric_to_check}"
self.lr_scheduler.step(metrics[metric_to_check])
if self.control.should_save:
self._save_checkpoint(model, trial, metrics=metrics)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment