Report Tune metrics in final evaluation (#7507)

dabc85d1 · Kai Fricke · GitHub · 9a92afb6 · dabc85d1
Unverified Commit dabc85d1 authored Oct 01, 2020 by Kai Fricke Committed by GitHub Oct 01, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

src/transformers/integrations.py src/transformers/integrations.py +1 -2

No files found.
--- a/src/transformers/integrations.py
+++ b/src/transformers/integrations.py
@@ -124,8 +124,7 @@ def run_hp_search_ray(trainer, n_trials: int, direction: str, **kwargs) -> BestR
            metrics = trainer.evaluate()
            trainer.objective = trainer.compute_objective(metrics)
            trainer._tune_save_checkpoint()
-            ray.tune.report(objective=trainer.objective)
+            ray.tune.report(objective=trainer.objective, **metrics, done=True)
-        return trainer.objective
    # The model and TensorBoard writer do not pickle so we have to remove them (if they exists)
    # while doing the ray hp search.