Lat fix for Ray HP search (#6691)

huggingface · Aug 24, 2020 · 8f98faf · 8f98faf
1 parent 3a7fdd3
commit 8f98faf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -843,6 +843,8 @@ def _objective(trial):
             if getattr(self, "objective", None) is None:
                 metrics = self.evaluate()
                 self.objective = self.compute_objective(metrics)
+                if self.hp_search_backend == HPSearchBackend.RAY:
+                    tune.report(objective=self.objective)
             return self.objective
 
         if self.hp_search_backend == HPSearchBackend.OPTUNA: