ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 10:42:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Revert Optuna deprecation fix (#613)

Browse Source
This commit is contained in:
allegroai 2022-03-24 19:23:29 +02:00
parent 6fb49fcd03
commit eed8f8cdfd
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/automation/optuna/optuna.py
View File

@ -1,4 +1,3 @@
from concurrent.futures import ThreadPoolExecutor
from time import sleep
from typing import Any, Optional, Sequence
@ -175,9 +174,8 @@ class OptimizerOptuna(SearchStrategy):
sleep_interval=int(self.pool_period_minutes * 60),
config_space=config_space,
)
with ThreadPoolExecutor(max_workers=self._num_concurrent_workers) as executor:
for _ in range(self._num_concurrent_workers):
executor.submit(self._study.optimize, self._objective.objective, n_trials=self.total_max_jobs)
self._study.optimize(
self._objective.objective, n_trials=self.total_max_jobs, n_jobs=self._num_concurrent_workers)
def stop(self):
# type: () -> ()