"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "e6c1f1cad84afe51019b90c169335612c3afc841"
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: ...@@ -2320,7 +2320,10 @@ class Trainer:
# Run delayed LR scheduler now that metrics are populated # Run delayed LR scheduler now that metrics are populated
if isinstance(self.lr_scheduler, torch.optim.lr_scheduler.ReduceLROnPlateau): 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: if self.control.should_save:
self._save_checkpoint(model, trial, metrics=metrics) 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