"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "5ef21866920e7353a5616b6a0eafc58a90a04824"
Unverified Commit 8f2c07d3 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix hp search for non sigopt backends (#13897)

parent 77770ec7
...@@ -1238,7 +1238,11 @@ class Trainer: ...@@ -1238,7 +1238,11 @@ class Trainer:
self.callback_handler.lr_scheduler = self.lr_scheduler self.callback_handler.lr_scheduler = self.lr_scheduler
self.callback_handler.train_dataloader = train_dataloader self.callback_handler.train_dataloader = train_dataloader
self.state.trial_name = self.hp_name(trial) if self.hp_name is not None else None self.state.trial_name = self.hp_name(trial) if self.hp_name is not None else None
self.state.trial_params = hp_params(trial.assignments) if trial is not None else None if trial is not None:
assignments = trial.assignments if self.hp_search_backend == HPSearchBackend.SIGOPT else trial
self.state.trial_params = hp_params(assignments)
else:
self.state.trial_params = None
# This should be the same if the state has been saved but in case the training arguments changed, it's safer # This should be the same if the state has been saved but in case the training arguments changed, it's safer
# to set this after the load. # to set this after the load.
self.state.max_steps = max_steps self.state.max_steps = max_steps
......
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