wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-05 13:21:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Save/Restore for Suggestion Algs (#5719)

Browse Source
This commit is contained in:
Hersh Godse
2019-09-21 11:11:57 -07:00
committed by Richard Liaw
parent 7e214fd95e
commit d17b35494d
6 changed files with 156 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/suggest/skopt.py
+7 -6
View File
@@ -160,11 +160,12 @@ class SkOptSearch(SuggestionAlgorithm):
return len(self._live_trial_mapping)
def save(self, checkpoint_dir):
trials_object = self._skopt_opt
with open(checkpoint_dir, "wb") as output:
pickle.dump(trials_object, output)
trials_object = (self._initial_points, self._skopt_opt)
with open(checkpoint_dir, "wb") as outputFile:
pickle.dump(trials_object, outputFile)
def restore(self, checkpoint_dir):
with open(checkpoint_dir, "rb") as input:
trials_object = pickle.load(input)
self._skopt_opt = trials_object
with open(checkpoint_dir, "rb") as inputFile:
trials_object = pickle.load(inputFile)
self._initial_points = trials_object[0]
self._skopt_opt = trials_object[1]