gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-19 21:42:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix up benchmark runner to use new benchmark format

Browse Source
This commit is contained in:
Jie Tang
2016-10-20 22:19:03 -07:00
parent a780b75556
commit c46008bfae
1 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
examples/scripts/benchmark_runner
View File

@@ -12,7 +12,7 @@ import os
import sys
import gym
import openai_benchmark
# In modules, use `logger = logging.getLogger(__name__)`
logger = logging.getLogger()
@@ -40,15 +40,14 @@ def main():
return 1
# run benchmark tasks
for env_name, task_list in benchmark.task_groups.items():
logger.info("Running on env: {}".format(env_name))
env = gym.make(env_name)
for task in task_list:
for seed in range(task.seeds):
training_dir_name = "{}/{}-{}".format(args.training_dir, env_name, seed)
env.monitor.start(training_dir_name, seed=seed)
for task in benchmark.tasks:
logger.info("Running on env: {}".format(task.env_id))
env = gym.make(task.env_id)
for trial in range(task.trials):
training_dir_name = "{}/{}-{}".format(args.training_dir, task.env_id, trial)
env.monitor.start(training_dir_name)
env.reset()
for _ in range(task.timesteps):
for _ in range(task.max_timesteps):
o, r, done, _ = env.step(env.action_space.sample())
if done:
env.reset()