From bc664c4169a3828cb6f99d3708f67fc0831dd965 Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Wed, 12 Sep 2018 19:53:04 +0000 Subject: [PATCH] add the first pass of rollout_worker.py --- rl_coach/rollout_worker.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 rl_coach/rollout_worker.py diff --git a/rl_coach/rollout_worker.py b/rl_coach/rollout_worker.py new file mode 100644 index 0000000..81f18ff --- /dev/null +++ b/rl_coach/rollout_worker.py @@ -0,0 +1,14 @@ +from rl_coach.base_parameters import TaskParameters +from rl_coach.core_types import EnvironmentEpisodes, RunPhase +from rl_coach.presets.CartPole_DQN import graph_manager + + +# TODO: workers might need to define schedules in terms which can be synchronized: exploration(len(distributed_memory)) -> float + +def main(): + graph_manager.create_graph(TaskParameters()) + graph_manager.phase = RunPhase.TRAIN + graph_manager.act(EnvironmentEpisodes(num_steps=10)) + +if __name__ == '__main__': + main()