ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-22 07:15:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix HPO crashes when optimizing for single value scalars (#1221)

Browse Source
This commit is contained in:
allegroai 2024-03-17 15:01:50 +02:00
parent 3531071fd5
commit f48d9d5a91
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
clearml/automation/optimization.py
View File

@ -2034,7 +2034,10 @@ class HyperParameterOptimizer(object):
if job_id in completed_jobs: if job_id in completed_jobs:
if value != completed_jobs[job_id][0]: if value != completed_jobs[job_id][0]:
iteration_value = self._objective_metric.get_current_raw_objective(job_id) iteration_value = self._objective_metric.get_current_raw_objective(job_id)
iteration = [it_[0] if it_ else -1 for it_ in iteration_value] if iteration_value:
iteration = [it_[0] if it_ else -1 for it_ in iteration_value]
else:
iteration = [-1]
completed_jobs[job_id] = ( completed_jobs[job_id] = (
value, value,
iteration, iteration,
@ -2049,7 +2052,10 @@ class HyperParameterOptimizer(object):
pairs.append((i, value)) pairs.append((i, value))
labels.append(str(params)[1:-1]) labels.append(str(params)[1:-1])
iteration_value = self._objective_metric.get_current_raw_objective(job_id) iteration_value = self._objective_metric.get_current_raw_objective(job_id)
iteration = [it_[0] if it_ else -1 for it_ in iteration_value] if iteration_value:
iteration = [it_[0] if it_ else -1 for it_ in iteration_value]
else:
iteration = [-1]
completed_jobs[job_id] = ( completed_jobs[job_id] = (
value, value,
iteration, iteration,