wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-10 12:30:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Fix Median Stopping Rule Verbosity (#1833)

Browse Source
This commit is contained in:
Richard Liaw
2018-04-06 22:58:13 -07:00
committed by GitHub
parent bef1d872b4
commit bc8f62c947
2 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/async_hyperband.py
+9 -7
View File
@@ -66,15 +66,17 @@ class AsyncHyperBandScheduler(FIFOScheduler):
self._trial_info[trial.trial_id] = self._brackets[idx]
def on_trial_result(self, trial_runner, trial, result):
action = TrialScheduler.CONTINUE
if getattr(result, self._time_attr) >= self._max_t:
action = TrialScheduler.STOP
else:
bracket = self._trial_info[trial.trial_id]
action = bracket.on_result(
trial,
getattr(result, self._time_attr),
getattr(result, self._reward_attr))
if action == TrialScheduler.STOP:
self._num_stopped += 1
return TrialScheduler.STOP
bracket = self._trial_info[trial.trial_id]
action = bracket.on_result(
trial,
getattr(result, self._time_attr),
getattr(result, self._reward_attr))
return action
def on_trial_complete(self, trial_runner, trial, result):