Lat fix for Ray HP search (huggingface#6691)

fabiocapsouza · Nov 15, 2020 · b387c1b · b387c1b
1 parent 621ce9b
commit b387c1b
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: