hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[tune] Allow to set buffer_length via tune.run (#15810)

Browse source
This commit is contained in:
lanlin 2021-05-17 20:11:26 +08:00 • committed by GitHub
parent 5247c0a5b8
commit 5d2ed47978
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/ray/tune/ray_trial_executor.py
View file

@ -142,6 +142,7 @@ class RayTrialExecutor(TrialExecutor):
def __init__(self,
queue_trials: bool = False,
reuse_actors: bool = False,
result_buffer_length: Optional[int] = None,
refresh_period: Optional[float] = None,
wait_for_placement_group: Optional[float] = None):
super(RayTrialExecutor, self).__init__(queue_trials)
@ -184,7 +185,8 @@ class RayTrialExecutor(TrialExecutor):
self.pg_recon_interval = float(
os.environ.get("TUNE_PLACEMENT_GROUP_RECON_INTERVAL", "5"))
self._buffer_length = int(os.getenv("TUNE_RESULT_BUFFER_LENGTH", 1000))
self._buffer_length = result_buffer_length or int(
os.getenv("TUNE_RESULT_BUFFER_LENGTH", 1000))
self._buffer_min_time_s = float(
os.getenv("TUNE_RESULT_BUFFER_MIN_TIME_S", 0.))
self._buffer_max_time_s = float(