cleanrl/benchmark/ppo_trxl.sh at master · vwxyzjn/cleanrl · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# export WANDB_ENTITY=openrlbenchmark

cd cleanrl/ppo_trxl
poetry install
OMP_NUM_THREADS=4 uv run python -m cleanrl_utils.benchmark \
    --env-ids MortarMayhem-Grid-v0 \
    --command "python ./cleanrl/ppo_trxl/ppo_trxl.py --track --norm_adv --trxl_memory_length 119 --total_timesteps 100000000" \
    --num-seeds 3 \
    --workers 32 \
    --slurm-template-path benchmark/cleanrl_1gpu.slurm_template

OMP_NUM_THREADS=4 uv run python -m cleanrl_utils.benchmark \
    --env-ids MortarMayhem-v0 \
    --command "python ./cleanrl/ppo_trxl/ppo_trxl.py --track --reconstruction_coef 0.1 --trxl_memory_length 275" \
    --num-seeds 3 \
    --workers 32 \
    --slurm-template-path benchmark/cleanrl_1gpu.slurm_template

OMP_NUM_THREADS=4 uv run python -m cleanrl_utils.benchmark \
    --env-ids MysteryPath-Grid-v0 \
    --command "python ./cleanrl/ppo_trxl/ppo_trxl.py --track --trxl_memory_length 96 --total_timesteps 100000000" \
    --num-seeds 3 \
    --workers 32 \
    --slurm-template-path benchmark/cleanrl_1gpu.slurm_template

OMP_NUM_THREADS=4 uv run python -m cleanrl_utils.benchmark \
    --env-ids MysteryPath-v0 \
    --command "python ./cleanrl/ppo_trxl/ppo_trxl.py --track --trxl_memory_length 256" \
    --num-seeds 3 \
    --workers 32 \
    --slurm-template-path benchmark/cleanrl_1gpu.slurm_template

OMP_NUM_THREADS=4 uv run python -m cleanrl_utils.benchmark \
    --env-ids SearingSpotlights-v0 \
    --command "python ./cleanrl/ppo_trxl/ppo_trxl.py --track --reconstruction_coef 0.1 --trxl_memory_length 256" \
    --num-seeds 3 \
    --workers 32 \
    --slurm-template-path benchmark/cleanrl_1gpu.slurm_template

OMP_NUM_THREADS=4 uv run python -m cleanrl_utils.benchmark \
    --env-ids Endless-SearingSpotlights-v0 \
    --command "python ./cleanrl/ppo_trxl/ppo_trxl.py --track --reconstruction_coef 0.1 --trxl_memory_length 256 --total_timesteps 350000000" \
    --num-seeds 3 \
    --workers 32 \
    --slurm-template-path benchmark/cleanrl_1gpu.slurm_template

OMP_NUM_THREADS=4 uv run python -m cleanrl_utils.benchmark \
    --env-ids Endless-MortarMayhem-v0 Endless-MysteryPath-v0 \
    --command "python ./cleanrl/ppo_trxl/ppo_trxl.py --track --trxl_memory_length 256 --total_timesteps 350000000" \
    --num-seeds 3 \
    --workers 32 \
    --slurm-template-path benchmark/cleanrl_1gpu.slurm_template