From 98142ef51f5291bf1304d86b0e05208a0ea8f054 Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Sat, 16 Sep 2017 15:29:36 -0700 Subject: [PATCH] fix checkpoint (#988) --- python/ray/rllib/common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/ray/rllib/common.py b/python/ray/rllib/common.py index 9c98bc52a..8c003316e 100644 --- a/python/ray/rllib/common.py +++ b/python/ray/rllib/common.py @@ -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."""