From 1a1730761af4e6314225477fc3770fe82b85bb09 Mon Sep 17 00:00:00 2001 From: Artur Mostowski Date: Mon, 1 Feb 2021 20:02:29 +0100 Subject: [PATCH 1/2] specify the list of players - only trains --- common/vec_env/subproc_vec_env.py | 1 + train_defeat_zerglings.py | 1 + 2 files changed, 2 insertions(+) diff --git a/common/vec_env/subproc_vec_env.py b/common/vec_env/subproc_vec_env.py index 2952ba0..0259c28 100644 --- a/common/vec_env/subproc_vec_env.py +++ b/common/vec_env/subproc_vec_env.py @@ -23,6 +23,7 @@ def worker(remote, map_name, nscripts, i): with sc2_env.SC2Env( agent_interface_format=[agent_format], map_name=map_name, + players=[sc2_env.Agent(sc2_env.Race.terran)], step_mul=2) as env: available_actions = [] result = None diff --git a/train_defeat_zerglings.py b/train_defeat_zerglings.py index 0fba7e3..bcadbbf 100644 --- a/train_defeat_zerglings.py +++ b/train_defeat_zerglings.py @@ -75,6 +75,7 @@ def main(): map_name="DefeatZerglingsAndBanelings", step_mul=step_mul, visualize=True, + players=[sc2_env.Agent(sc2_env.Race.terran)], game_steps_per_episode=steps * step_mul) as env: model = deepq.models.cnn_to_mlp( From 6cb76208f0d440e15e90b7ffdcd61ecd258bfed4 Mon Sep 17 00:00:00 2001 From: Artur Mostowski Date: Mon, 1 Feb 2021 20:15:03 +0100 Subject: [PATCH 2/2] agent_interface_format - train defeat zerglings --- train_defeat_zerglings.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/train_defeat_zerglings.py b/train_defeat_zerglings.py index bcadbbf..f144540 100644 --- a/train_defeat_zerglings.py +++ b/train_defeat_zerglings.py @@ -76,6 +76,10 @@ def main(): step_mul=step_mul, visualize=True, players=[sc2_env.Agent(sc2_env.Race.terran)], + agent_interface_format=sc2_env.AgentInterfaceFormat( + feature_dimensions=sc2_env.Dimensions( + screen=64, + minimap=64)), game_steps_per_episode=steps * step_mul) as env: model = deepq.models.cnn_to_mlp(