wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-15 11:25:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Allow relative local_dir at tune.run() (#4734)

Browse Source
This commit is contained in:
Peng Zhenghao
2019-08-10 16:49:34 -07:00
committed by Richard Liaw
parent a1d2e17623
commit 983f3c83d8
3 changed files with 171 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/trainable.py
+8
View File
@@ -350,6 +350,14 @@ class Trainable(object):
self._timesteps_since_restore = 0
self._iterations_since_restore = 0
self._restored = True
logger.info("Restored from checkpoint: {}".format(checkpoint_path))
state = {
"_iteration": self._iteration,
"_timesteps_total": self._timesteps_total,
"_time_total": self._time_total,
"_episodes_total": self._episodes_total,
}
logger.info("Current state after restoring: {}".format(state))
def restore_from_object(self, obj):
"""Restores training state from a checkpoint object.