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

Adding worker logs and plumbed task_parameters to distributed coach (#130)

Browse Source
This commit is contained in:
Ajay Deshpande
2018-11-23 15:35:11 -08:00
committed by Balaji Subramaniam
parent 2b4c9c6774
commit 4a6c404070
5 changed files with 84 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rl_coach/training_worker.py
View File

@@ -12,14 +12,11 @@ def data_store_ckpt_save(data_store):
time.sleep(10)
def training_worker(graph_manager, checkpoint_dir):
def training_worker(graph_manager, task_parameters):
"""
restore a checkpoint then perform rollouts using the restored model
"""
# initialize graph
task_parameters = TaskParameters()
task_parameters.__dict__['checkpoint_save_dir'] = checkpoint_dir
task_parameters.__dict__['checkpoint_save_secs'] = 20
graph_manager.create_graph(task_parameters)
# save randomly initialized graph