ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add HyperParameterOptimizer arguments: spawn_task_project, save_top_k_tasks_only

Browse Source
This commit is contained in:
allegroai
2021-01-10 13:07:57 +02:00
parent 49ba89587f
commit f679c80535
2 changed files with 233 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/optimization/hyper-parameter-optimization/hyper_parameter_optimizer.py
View File

@@ -86,6 +86,10 @@ an_optimizer = HyperParameterOptimizer(
optimizer_class=aSearchStrategy,
# Select an execution queue to schedule the experiments for execution
execution_queue=execution_queue,
# If specified all Tasks created by the HPO process will be under the `spawned_tasks_project` project
spawn_task_project=None, # 'HPO spawn project',
# If specified only the top K performing Tasks will be kept, the others will be automatically archived
save_top_k_tasks_only=None, # 5,
# Optional: Limit the execution time of a single experiment, in minutes.
# (this is optional, and if using OptimizerBOHB, it is ignored)
time_limit_per_job=10.,