NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update hpo.py

Browse source
This commit is contained in:
crStiv 2026-01-15 16:05:27 +02:00 • committed by GitHub
parent d2fbe43e7e
commit 20992ed5d5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
environments/community/pytorch_optimizer_coding/FOB/examples/neps/hpo.py
View file

@ -70,7 +70,7 @@ def create_experiment(run: Run, config: dict) -> dict:
# Define the run pipeline function
def create_pipline(base_run: Run):
def create_pipeline(base_run: Run):
def run_pipeline(pipeline_directory, previous_pipeline_directory, **config) -> dict:
# Initialize the first directory to store the event and checkpoints files
init_dir = get_initial_directory(pipeline_directory)
@ -183,7 +183,7 @@ if __name__ == "__main__":
# Run NePS with specified parameters
neps.run(
run_pipeline=create_pipline(run),
run_pipeline=create_pipeline(run),
pipeline_space=search_space(run),
root_directory=run.engine.output_dir,
max_evaluations_total=args.n_trials,