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

fix simple typos; temporarily disable redis in rollout worker

Browse Source
This commit is contained in:
Zach Dwiel
2018-09-17 18:33:20 +00:00
committed by zach dwiel
parent f5b7122d56
commit 009cf670f3
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rl_coach/rollout_worker.py
View File

@@ -9,6 +9,7 @@ this rollout worker:
import argparse import argparse
import time import time
import os
from rl_coach.base_parameters import TaskParameters from rl_coach.base_parameters import TaskParameters
from rl_coach.coach import expand_preset from rl_coach.coach import expand_preset
@@ -39,7 +40,9 @@ def wait_for_checkpoint(checkpoint_dir, timeout=10):
if has_checkpoint(checkpoint_dir): if has_checkpoint(checkpoint_dir):
return return
raise ValueError(f'checkpoint never found in {checkpoint_dir}') raise ValueError('checkpoint never found in {checkpoint_dir}'.format(
checkpoint_dir=checkpoint_dir,
))
def rollout_worker(graph_manager, checkpoint_dir): def rollout_worker(graph_manager, checkpoint_dir):
@@ -79,8 +82,9 @@ def main():
graph_manager = short_dynamic_import(expand_preset(args.preset), ignore_module_case=True) graph_manager = short_dynamic_import(expand_preset(args.preset), ignore_module_case=True)
graph_manager.agent_params.memory.redis_ip = args.redis_ip # TODO: get this working, this expects that memory already has a redis ip and port
graph_manager.agent_params.memory.redis_port = args.redis_port # graph_manager.agent_params.memory.redis_ip = args.redis_ip
# graph_manager.agent_params.memory.redis_port = args.redis_port
rollout_worker( rollout_worker(
graph_manager=graph_manager, graph_manager=graph_manager,