gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-09 15:22:36 +02:00
Code Issues Projects Releases Wiki Activity

rename save_checkpoint_dir -> checkpoint_save_dir

Browse Source
This commit is contained in:
Zach Dwiel authored and zach dwiel committed 2018-10-23 17:10:58 -04:00
1 parent 201a2237a1
commit 9804b033a2
4 files changed
+5 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

rl_coach/graph_managers/graph_manager.py
+2 -2
View File
@@ -203,7 +203,7 @@ class GraphManager(object):
remove_tree(checkpoint_dir)
copy_tree(task_parameters.checkpoint_restore_dir, checkpoint_dir)
else:
checkpoint_dir = task_parameters.save_checkpoint_dir
checkpoint_dir = task_parameters.checkpoint_save_dir
self.sess = create_monitored_session(target=task_parameters.worker_target,
task_index=task_parameters.task_index,
@@ -498,7 +498,7 @@ class GraphManager(object):
self.save_checkpoint()
def save_checkpoint(self):
checkpoint_path = os.path.join(self.task_parameters.save_checkpoint_dir,
checkpoint_path = os.path.join(self.task_parameters.checkpoint_save_dir,
"{}_Step-{}.ckpt".format(
self.checkpoint_id,
self.total_steps_counters[RunPhase.TRAIN][EnvironmentSteps]))