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

fixed bug in training worker

Browse Source
This commit is contained in:
Zach Dwiel
2018-09-20 11:26:17 -04:00
committed by zach dwiel
parent cd733b2404
commit d69332efd4
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rl_coach/training_worker.py
View File

@@ -18,9 +18,8 @@ def heatup(graph_manager):
redis_ip=graph_manager.agent_params.memory.redis_ip, redis_ip=graph_manager.agent_params.memory.redis_ip,
redis_port=graph_manager.agent_params.memory.redis_port) redis_port=graph_manager.agent_params.memory.redis_port)
num_steps = graph_manager.schedule_params.heatup_steps.num_steps while(memory.num_transitions() < graph_manager.heatup_steps.num_steps):
while(memory.num_transitions() < num_steps): time.sleep(1)
time.sleep(10)
def training_worker(graph_manager, checkpoint_dir): def training_worker(graph_manager, checkpoint_dir):
@@ -48,6 +47,7 @@ def training_worker(graph_manager, checkpoint_dir):
graph_manager.save_checkpoint() graph_manager.save_checkpoint()
# TODO: signal to workers that training is done
def main(): def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()