wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 20:07:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] logging fixes, better warnings, better cluster support (#3906)

Browse Source
This commit is contained in:
Richard Liaw
2019-02-02 19:14:03 -08:00
committed by GitHub
parent 002531b199
commit eab6dd72b5
5 changed files with 32 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/function_runner.py
+6
View File
@@ -32,11 +32,14 @@ class StatusReporter(object):
def __call__(self, **kwargs):
"""Report updated training status.
Pass in `done=True` when the training job is completed.
Args:
kwargs: Latest training result status.
Example:
>>> reporter(mean_accuracy=1, training_iteration=4)
>>> reporter(mean_accuracy=1, training_iteration=4, done=True)
"""
with self._lock:
@@ -48,6 +51,9 @@ class StatusReporter(object):
if self._done and not self._latest_result:
if not self._last_result:
raise TuneError("Trial finished without reporting result!")
logger.warning("Trial detected as completed; re-reporting "
"last result. To avoid this, include done=True "
"upon the last reporter call.")
self._last_result.update(done=True)
return self._last_result
with self._lock: