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(