From 00e0f14c6fb5be126c3532a0b8dbaf46776b04ce Mon Sep 17 00:00:00 2001 From: Vishnu Deva Date: Thu, 15 Oct 2020 07:39:34 +0530 Subject: [PATCH] [tune] restore trials when `sync_on_checkpoint` is False (#11355) Co-authored-by: Richard Liaw --- python/ray/tune/ray_trial_executor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/ray/tune/ray_trial_executor.py b/python/ray/tune/ray_trial_executor.py index bbe2def83..1ed639678 100644 --- a/python/ray/tune/ray_trial_executor.py +++ b/python/ray/tune/ray_trial_executor.py @@ -722,7 +722,8 @@ class RayTrialExecutor(TrialExecutor): trial.runner.restore_from_object.remote(value) else: logger.debug("Trial %s: Attempting restore from %s", trial, value) - if issubclass(trial.get_trainable_cls(), DurableTrainable): + if issubclass(trial.get_trainable_cls(), + DurableTrainable) or not trial.sync_on_checkpoint: with self._change_working_directory(trial): remote = trial.runner.restore.remote(value) elif trial.sync_on_checkpoint: