From 51562f3e0b200a46ea64ff06f88544d8a4fa8d0a Mon Sep 17 00:00:00 2001 From: vmoens Date: Mon, 9 Oct 2023 20:58:53 +0100 Subject: [PATCH] init --- benchmarks/ecosystem/gym_env_throughput.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/benchmarks/ecosystem/gym_env_throughput.py b/benchmarks/ecosystem/gym_env_throughput.py index 457f15a2b5a..146d011442d 100644 --- a/benchmarks/ecosystem/gym_env_throughput.py +++ b/benchmarks/ecosystem/gym_env_throughput.py @@ -30,17 +30,14 @@ if __name__ == "__main__": for envname in [ - "HalfCheetah-v4", "CartPole-v1", + "HalfCheetah-v4", "myoHandReachRandom-v0", "ALE/Breakout-v5", - "CartPole-v1", ]: # the number of collectors won't affect the resources, just impacts how the envs are split in sub-sub-processes - for num_workers, num_collectors in zip((8, 16, 32, 64), (2, 4, 8, 8)): - with open( - f"atari_{envname}_{num_workers}.txt".replace("/", "-"), "w+" - ) as log: + for num_workers, num_collectors in zip((32, 64, 8, 16), (8, 8, 2, 4)): + with open(f"{envname}_{num_workers}.txt".replace("/", "-"), "w+") as log: if "myo" in envname: gym_backend = "gym" else: @@ -219,7 +216,7 @@ def make_env( penv = EnvCreator( lambda num_workers=num_workers // num_collectors: make_env( - num_workers + num_workers=num_workers ) ) collector = MultiaSyncDataCollector( @@ -306,7 +303,7 @@ def make_env( penv = EnvCreator( lambda num_workers=num_workers // num_collectors: make_env( - num_workers + num_workers=num_workers ) ) collector = MultiSyncDataCollector(