Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "fb056494e50be6ff729166734eff96f9f26d9e96"
Unverified Commit 8f98faf9 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Lat fix for Ray HP search (#6691)

parent 3a7fdd3f
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/transformers/trainer.py
View file @ 8f98faf9
...@@ -843,6 +843,8 @@ class Trainer: ...@@ -843,6 +843,8 @@ class Trainer:
if getattr(self, "objective", None) is None: if getattr(self, "objective", None) is None:
metrics = self.evaluate() metrics = self.evaluate()
self.objective = self.compute_objective(metrics) self.objective = self.compute_objective(metrics)
if self.hp_search_backend == HPSearchBackend.RAY:
tune.report(objective=self.objective)
return self.objective return self.objective
if self.hp_search_backend == HPSearchBackend.OPTUNA: if self.hp_search_backend == HPSearchBackend.OPTUNA:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment