diff --git a/tuning/hp_search_spaces.py b/tuning/hp_search_spaces.py index 196397cfa..3beed298c 100644 --- 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(