From ffa043d4b775c61764ae217b54ae989d893e46e1 Mon Sep 17 00:00:00 2001 From: Yuhao Yang Date: Sat, 30 Nov 2019 03:09:30 +0800 Subject: [PATCH] [tune] replace self.config (#6313) --- python/ray/tune/examples/pbt_example.py | 1 + python/ray/tune/trainable.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/python/ray/tune/examples/pbt_example.py b/python/ray/tune/examples/pbt_example.py index 96a8a4ec4..c3acecdab 100755 --- a/python/ray/tune/examples/pbt_example.py +++ b/python/ray/tune/examples/pbt_example.py @@ -81,6 +81,7 @@ class PBTBenchmarkExample(Trainable): def reset_config(self, new_config): self.lr = new_config["lr"] + self.config = new_config return True diff --git a/python/ray/tune/trainable.py b/python/ray/tune/trainable.py index 988b4b6c5..a6b668de2 100644 --- a/python/ray/tune/trainable.py +++ b/python/ray/tune/trainable.py @@ -396,7 +396,8 @@ class Trainable(object): This method is optional, but can be implemented to speed up algorithms such as PBT, and to allow performance optimizations such as running - experiments with reuse_actors=True. + experiments with reuse_actors=True. Note that self.config need to + be updated to reflect the latest parameter information in Ray logs. Args: new_config (dir): Updated hyperparameter configuration