Store command name as user attr in trial.

HumanCompatibleAI · Feb 27, 2024 · 9bb89c9 · 9bb89c9
1 parent 80edc62
commit 9bb89c9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tuning/hp_search_spaces.py b/tuning/hp_search_spaces.py
@@ -64,6 +64,7 @@ def __call__(
 
  trial.set_user_attr("config_updates", config_updates)
  trial.set_user_attr("named_configs", named_configs)
+ trial.set_user_attr("command_name", self.command_name)
 
  experiment: sacred.Experiment = self.sacred_ex
  result = experiment.run(