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

update training worker

Browse Source
This commit is contained in:
zach dwiel
2018-10-11 12:29:57 -04:00
parent 430ca198e5
commit 3e5e5475de
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/training_worker.py
View File

@@ -25,7 +25,7 @@ def training_worker(graph_manager, checkpoint_dir, policy_type):
# initialize graph # initialize graph
task_parameters = TaskParameters() task_parameters = TaskParameters()
task_parameters.__dict__['checkpoint_save_dir'] = checkpoint_dir task_parameters.__dict__['checkpoint_save_dir'] = checkpoint_dir
task_parameters.__dict__['save_checkpoint_secs'] = 20 task_parameters.__dict__['checkpoint_save_secs'] = 20
graph_manager.create_graph(task_parameters) graph_manager.create_graph(task_parameters)
# save randomly initialized graph # save randomly initialized graph