mirror of
https://github.com/wassname/ray.git
synced 2026-07-24 13:20:22 +08:00
[tune] stdout/stderr logging redirection (#9817)
* Add `log_to_file` parameter, pass to Trainable config, redirect stdout/stderr. * Add logging handler to root ray logger * Added test for `log_to_file` parameter * Added logs, reuse test * Revert debug change * Update logdir on reset, flush streams after each train() step * Remove magic keys from visible config Co-authored-by: Kai Fricke <kai@anyscale.com>
This commit is contained in:
@@ -137,6 +137,20 @@ class warn_if_slow:
|
||||
now - self.start)
|
||||
|
||||
|
||||
class Tee(object):
|
||||
def __init__(self, stream1, stream2):
|
||||
self.stream1 = stream1
|
||||
self.stream2 = stream2
|
||||
|
||||
def write(self, *args, **kwargs):
|
||||
self.stream1.write(*args, **kwargs)
|
||||
self.stream2.write(*args, **kwargs)
|
||||
|
||||
def flush(self, *args, **kwargs):
|
||||
self.stream1.flush(*args, **kwargs)
|
||||
self.stream2.flush(*args, **kwargs)
|
||||
|
||||
|
||||
def merge_dicts(d1, d2):
|
||||
"""
|
||||
Args:
|
||||
|
||||
Reference in New Issue
Block a user