wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-27 22:08:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] restore trials when sync_on_checkpoint is False (#11355)

Browse Source 
Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Vishnu Deva
2020-10-15 07:39:34 +05:30
committed by GitHub
parent f7120d2a18
commit 00e0f14c6f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/ray_trial_executor.py
+2 -1
View File
@@ -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: