Skip to content

Commit

Permalink
add commment to multi fidelity example for clarification
Browse files Browse the repository at this point in the history
  • Loading branch information
danrgll committed Jan 25, 2024
1 parent c3596ef commit 3160b43
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions neps_examples/efficiency/multi_fidelity.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,11 @@ def get_model_and_optimizer(learning_rate):
return model, optimizer


# Important: Include the "pipeline_directory" and "previous_pipeline_directory" arguments
# in your run_pipeline function. This grants access to NePS's folder system and is
# critical for leveraging efficient multi-fidelity optimization strategies.


def run_pipeline(pipeline_directory, previous_pipeline_directory, learning_rate, epoch):
model, optimizer = get_model_and_optimizer(learning_rate)
checkpoint_name = "checkpoint.pth"
Expand Down

0 comments on commit 3160b43

Please sign in to comment.