Small fixes to HP search (#7839)

2020-10-16 03:23:44 -04:00
parent d8ca57d2ce
commit eb186bc14e
2 changed files with 2 additions and 1 deletions
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -520,7 +520,7 @@ class Trainer:
    ):
        if self.hp_search_backend is None or trial is None:
            return
-        self.objective = self.compute_objective(metrics)
+        self.objective = self.compute_objective(metrics.copy())
        if self.hp_search_backend == HPSearchBackend.OPTUNA:
            trial.report(self.objective, epoch)
            if trial.should_prune():