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

Adding kubernetes orchestrator for rollouts, adding requirements for incremental docker builds

Browse Source
This commit is contained in:
Ajay Deshpande
2018-09-14 15:58:57 -07:00
committed by zach dwiel
parent 6541bc76b9
commit ce9838a7d6
10 changed files with 327 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/rollout_worker.py
View File

@@ -20,6 +20,7 @@ def rollout_worker(graph_manager, checkpoint_dir):
task_parameters = TaskParameters()
task_parameters.__dict__['checkpoint_restore_dir'] = checkpoint_dir
graph_manager.create_graph(task_parameters)
graph_manager.phase = RunPhase.TRAIN
graph_manager.act(EnvironmentEpisodes(num_steps=10))
graph_manager.phase = RunPhase.UNDEFINED
@@ -44,5 +45,6 @@ def main():
checkpoint_dir=args.checkpoint_dir,
)
if __name__ == '__main__':
main()