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

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

2
rl_coach/training_worker.py
View File

@@ -24,7 +24,7 @@ def training_worker(graph_manager, checkpoint_dir, policy_type):
"""
# initialize graph
task_parameters = TaskParameters()
task_parameters.__dict__['save_checkpoint_dir'] = checkpoint_dir
task_parameters.__dict__['checkpoint_save_dir'] = checkpoint_dir
task_parameters.__dict__['save_checkpoint_secs'] = 20
graph_manager.create_graph(task_parameters)