gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

bug-fix for checkpointing for single-worker algorithms

Browse Source
This commit is contained in:
Gal Leibovich
2018-08-19 20:17:15 +03:00
parent 9f599f38cf
commit c1f428666e
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rl_coach/graph_managers/graph_manager.py
View File

@@ -426,8 +426,10 @@ class GraphManager(object):
def save_checkpoint(self): def save_checkpoint(self):
# only the chief process saves checkpoints # only the chief process saves checkpoints
if self.task_parameters.save_checkpoint_secs \ if self.task_parameters.save_checkpoint_secs \
and time.time() - self.last_checkpoint_saving_time >= self.task_parameters.save_checkpoint_secs\ and time.time() - self.last_checkpoint_saving_time >= self.task_parameters.save_checkpoint_secs \
and self.task_parameters.task_index == 0: and (self.task_parameters.task_index == 0 # distributed
or self.task_parameters.task_index is None # single-worker
):
checkpoint_path = os.path.join(self.task_parameters.save_checkpoint_dir, checkpoint_path = os.path.join(self.task_parameters.save_checkpoint_dir,
"{}_Step-{}.ckpt".format( "{}_Step-{}.ckpt".format(