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

fix checkpoint (#988)

Browse Source
This commit is contained in:
Eric Liang
2017-09-16 15:29:36 -07:00
committed by Philipp Moritz
parent 241612709e
commit 98142ef51f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/common.py
+2 -2
View File
@@ -170,7 +170,7 @@ class Agent(object):
checkpoint_path = self._save()
pickle.dump(
[self.experiment_id, self.iteration, self.timesteps_total,
self.time_total_s],
self.time_total],
open(checkpoint_path + ".rllib_metadata", "wb"))
return checkpoint_path
@@ -185,7 +185,7 @@ class Agent(object):
self.experiment_id = metadata[0]
self.iteration = metadata[1]
self.timesteps_total = metadata[2]
self.time_total_s = metadata[3]
self.time_total = metadata[3]
def compute_action(self, observation):
"""Computes an action using the current trained policy."""