gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-20 05:52:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make random benchmark runner script respect seeds

Browse Source
This commit is contained in:
Jie Tang
2016-10-17 11:42:42 -07:00
parent 65b3454a77
commit 8941313121
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
examples/scripts/benchmark_runner
View File

@@ -43,14 +43,16 @@ def main():
for env_name, task_list in benchmark.task_groups.items():
logger.info("Running on env: {}".format(env_name))
env = gym.make(env_name)
env.monitor.start("{}/{}".format(args.training_dir, env_name))
for task in task_list:
env.reset()
for _ in range(task.timesteps):
o, r, done, _ = env.step(env.action_space.sample())
if done:
env.reset()
env.monitor.close()
for seed in range(task.seeds):
training_dir_name = "{}/{}-{}".format(args.training_dir, env_name, seed)
env.monitor.start(training_dir_name, seed=seed)
env.reset()
for _ in range(task.timesteps):
o, r, done, _ = env.step(env.action_space.sample())
if done:
env.reset()
env.monitor.close()
logger.info("""Done running, upload results using the following command: