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

Implement frame-work agnostic rollout and training workers (#137)

Browse Source 
* Added checkpoint state file to coach checkpointing.

* Removed TF specific code from rollout_worker, training_worker, and s3_data_store
This commit is contained in:
Sina Afrooze
2018-11-23 18:05:44 -08:00
committed by Balaji Subramaniam
parent 4a6c404070
commit 5332013bd1
7 changed files with 350 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/training_worker.py
View File

@@ -30,7 +30,7 @@ def training_worker(graph_manager, task_parameters):
graph_manager.setup_memory_backend()
while(steps < graph_manager.improve_steps.num_steps):
while steps < graph_manager.improve_steps.num_steps:
graph_manager.phase = core_types.RunPhase.TRAIN
graph_manager.fetch_from_worker(graph_manager.agent_params.algorithm.num_consecutive_playing_steps)