gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 03:30:19 +01:00
Code Issues Projects Releases Wiki Activity

Adding steps and waiting for new checkpoint

Browse Source
This commit is contained in:
Ajay Deshpande
2018-10-08 13:41:51 -07:00
committed by zach dwiel
parent 0e121c5762
commit 0f46877d7e
3 changed files with 23 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/training_worker.py
View File

@@ -31,8 +31,10 @@ def training_worker(graph_manager, checkpoint_dir):
graph_manager.save_checkpoint()
# training loop
while True:
steps = 0
while(steps < graph_manager.improve_steps.num_steps):
if graph_manager.should_train():
steps += 1
graph_manager.phase = core_types.RunPhase.TRAIN
graph_manager.train(core_types.TrainingSteps(1))
graph_manager.phase = core_types.RunPhase.UNDEFINED