Lat fix for Ray HP search (#6691)

8f98faf9 · Sylvain Gugger · GitHub · 3a7fdd3f · 8f98faf9
Unverified Commit 8f98faf9 authored Aug 24, 2020 by Sylvain Gugger Committed by GitHub Aug 24, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/transformers/trainer.py src/transformers/trainer.py +2 -0

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -843,6 +843,8 @@ class Trainer:
            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: