<filename>runner/runs/voxel_performance_benchmark_all_envs.py<gh_stars>1-10
from runner.run_description import RunDescription, Experiment, ParamGrid

NUM_WORKERS_VOXEL_ENV = 48  # typically num logical cores / 2, limited by the num of available Vulkan contexts
TIMEOUT_SECONDS = 180
SAMPLER_GPUS = '0 1 2 3 4 5 6 7'  # replace with '0 1 2 3 4 5 6 7' for 8-GPU server

_voxel_env_cli = f'python -m run_algorithm --algo=DUMMY_SAMPLER --num_workers={NUM_WORKERS_VOXEL_ENV} --num_envs_per_worker=1 --experiment=benchmark --sampler_worker_gpus {SAMPLER_GPUS} --voxel_num_envs_per_instance=64 --voxel_num_agents_per_env=2 --voxel_num_simulation_threads=2 --timeout_seconds={TIMEOUT_SECONDS}'

_params_voxel_env = ParamGrid([
    ('env',
     [
         'voxel_env_TowerBuilding',
         'voxel_env_ObstaclesEasy',
         'voxel_env_ObstaclesHard',
         'voxel_env_Collect',
         'voxel_env_Sokoban',
         'voxel_env_HexMemory',
         'voxel_env_HexExplore',
         'voxel_env_Rearrange'
     ]),
    ('voxel_use_vulkan', [True]),
])

_experiment_voxel_env = Experiment(
    'benchmark_voxel_env_8',
    _voxel_env_cli,
    _params_voxel_env.generate_params(randomize=False),
)

RUN_DESCRIPTION = RunDescription('voxel_bench_sampling_all_envs',
                                 experiments=[_experiment_voxel_env])

<filename>runner/runs/voxel_performance_benchmark_all_envs.py<gh_stars>1-10
from runner.run_description import RunDescription, Experiment, ParamGrid

NUM_WORKERS_VOXEL_ENV = 48  # typically num logical cores / 2, limited by the num of available Vulkan contexts
TIMEOUT_SECONDS = 180
SAMPLER_GPUS = '0 1 2 3 4 5 6 7'  # replace with '0 1 2 3 4 5 6 7' for 8-GPU server

_voxel_env_cli = f'python -m run_algorithm --algo=DUMMY_SAMPLER --num_workers={NUM_WORKERS_VOXEL_ENV} --num_envs_per_worker=1 --experiment=benchmark --sampler_worker_gpus {SAMPLER_GPUS} --voxel_num_envs_per_instance=64 --voxel_num_agents_per_env=2 --voxel_num_simulation_threads=2 --timeout_seconds={TIMEOUT_SECONDS}'

_params_voxel_env = ParamGrid([
    ('env',
     [
         'voxel_env_TowerBuilding',
         'voxel_env_ObstaclesEasy',
         'voxel_env_ObstaclesHard',
         'voxel_env_Collect',
         'voxel_env_Sokoban',
         'voxel_env_HexMemory',
         'voxel_env_HexExplore',
         'voxel_env_Rearrange'
     ]),
    ('voxel_use_vulkan', [True]),
])

_experiment_voxel_env = Experiment(
    'benchmark_voxel_env_8',
    _voxel_env_cli,
    _params_voxel_env.generate_params(randomize=False),
)

RUN_DESCRIPTION = RunDescription('voxel_bench_sampling_all_envs',
                                 experiments=[_experiment_voxel_env])

# typically num logical cores / 2, limited by the num of available Vulkan contexts # replace with '0 1 2 3 4 5 6 7' for 8-GPU server