From 87f7ce5f8addf84d349c8a290e8dac165dd47752 Mon Sep 17 00:00:00 2001 From: Kaiyotech <93724202+Kaiyotech@users.noreply.github.com> Date: Tue, 7 Feb 2023 09:55:19 -0500 Subject: [PATCH 1/2] working 1v0 --- .../rollout_generator/redis/redis_rollout_generator.py | 3 ++- rocket_learn/rollout_generator/redis/redis_rollout_worker.py | 3 +++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/rocket_learn/rollout_generator/redis/redis_rollout_generator.py b/rocket_learn/rollout_generator/redis/redis_rollout_generator.py index de12d0b..e3288ab 100644 --- a/rocket_learn/rollout_generator/redis/redis_rollout_generator.py +++ b/rocket_learn/rollout_generator/redis/redis_rollout_generator.py @@ -312,7 +312,8 @@ def update_parameters(self, new_params): "redis/contributors": wandb.Table(columns=["name", "steps"], data=self.contributors.most_common())}, commit=False ) - self._plot_ratings() + if self.gamemodes[0] != '1v0': + self._plot_ratings() tot_contributors = self.redis.hgetall(CONTRIBUTORS) tot_contributors = Counter({name: int(count) for name, count in tot_contributors.items()}) tot_contributors += self.contributors diff --git a/rocket_learn/rollout_generator/redis/redis_rollout_worker.py b/rocket_learn/rollout_generator/redis/redis_rollout_worker.py index a29bfd3..5739ef3 100644 --- a/rocket_learn/rollout_generator/redis/redis_rollout_worker.py +++ b/rocket_learn/rollout_generator/redis/redis_rollout_worker.py @@ -254,6 +254,9 @@ def run(self): # Mimics Thread if self.dynamic_gm: blue, orange = self.select_gamemode() + elif self.match.agents == 1: + blue = 1 + orange = 0 else: blue = orange = self.match.agents // 2 self.set_team_size(blue, orange) From 6472dad29dd7ce5df1952757e454e001ed627855 Mon Sep 17 00:00:00 2001 From: Kaiyotech <93724202+Kaiyotech@users.noreply.github.com> Date: Tue, 7 Feb 2023 10:03:42 -0500 Subject: [PATCH 2/2] working 1v0 --- rocket_learn/rollout_generator/redis/redis_rollout_worker.py | 1 + 1 file changed, 1 insertion(+) diff --git a/rocket_learn/rollout_generator/redis/redis_rollout_worker.py b/rocket_learn/rollout_generator/redis/redis_rollout_worker.py index 5739ef3..e2c810b 100644 --- a/rocket_learn/rollout_generator/redis/redis_rollout_worker.py +++ b/rocket_learn/rollout_generator/redis/redis_rollout_worker.py @@ -117,6 +117,7 @@ def __init__(self, redis: Redis, name: str, match: Match, def _get_opponent_ids(self, n_new, n_old, pretrained_choice): # Get qualities gamemode = f"{(n_new + n_old) // 2}v{(n_new + n_old) // 2}" + gamemode = '1v0' if gamemode == '0v0' else gamemode latest_id = self.redis.get(LATEST_RATING_ID).decode("utf-8") latest_key = f"{latest_id}-stochastic" if n_old == 0: