Skip to content

Commit

Permalink
[add] docstring for new seed_config_auto argument
Browse files Browse the repository at this point in the history
  • Loading branch information
MaxFBurg committed May 30, 2022
1 parent 3e422c3 commit c2d81f1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nnfabrik/utility/hypersearch.py
Original file line number Diff line number Diff line change
Expand Up @@ -266,6 +266,8 @@ class Random:
trainer_fn (str): name of the trainer function
trainer_config (dict): dictionary of arguments for trainer function that are fixed
trainer_config_auto (dict): dictionary of arguments for trainer function that are to be randomly sampled
seed_config_auto (dict): dictionary of arguments for setting (`dict(seed={"type": "fixed", "value": <VALUE>})`)
or random sampling (`dict(seed={"type": "int"})`) the seed
architect (str): Name of the contributor that added this entry
trained_model_table (str): name (importable) of the trained_model_table
total_trials (int, optional): Number of experiments (i.e. training) to run. Defaults to 5.
Expand Down Expand Up @@ -499,7 +501,5 @@ def run(self):
"""
Runs the random hyperparameter search, for as many trials as specified.
"""
# n_trials = len(self.trained_model_table().seed_table()) * self.total_trials
# init_len = len(self.trained_model_table())
for _ in range(self.total_trials):
self.train_evaluate(self.gen_params_value())

0 comments on commit c2d81f1

Please sign in to comment.