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

rename save_checkpoint_dir -> checkpoint_save_dir

Browse Source
This commit is contained in:
Zach Dwiel
2018-10-05 11:44:49 -04:00
committed by zach dwiel
parent 201a2237a1
commit 9804b033a2
4 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/graph_managers/graph_manager.py
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]))