1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00

Adding right arguments to the agent

This commit is contained in:
Ajay Deshpande
2018-09-14 16:17:34 -07:00
committed by zach dwiel
parent ad7f031031
commit c2991819b4
2 changed files with 44 additions and 49 deletions

View File

@@ -40,6 +40,9 @@ def main():
graph_manager = short_dynamic_import(expand_preset(args.preset), ignore_module_case=True)
graph_manager.agent_parameters.memory.redis_ip = args.redis_ip
graph_manager.agent_params.memory.redis_port = args.redis_port
rollout_worker(
graph_manager=graph_manager,
checkpoint_dir=args.checkpoint_dir,