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: