Commit 00b5d4b5 authored by manueth's avatar manueth

test configs adjusted to improve speed, tests added to test.sh

parent dc367bae
Pipeline #4880 passed with stage
in 50 minutes and 20 seconds
......@@ -2,7 +2,7 @@ flatland-sparse-small-density-cnn-apex:
run: APEX
env: flatland_sparse
stop:
timesteps_total: 10000
timesteps_total: 500
checkpoint_freq: 10
checkpoint_at_end: True
keep_checkpoints_num: 5
......
......@@ -2,7 +2,7 @@ flatland-sparse-small-density-cnn-apex:
run: APEX
env: flatland_sparse
stop:
timesteps_total: 10000
timesteps_total: 500
checkpoint_freq: 10
checkpoint_at_end: True
keep_checkpoints_num: 5
......
......@@ -2,7 +2,7 @@ flatland-sparse-global-conv-ppo:
run: PPO
env: flatland_sparse
stop:
timesteps_total: 500 # 1e7
timesteps_total: 500
checkpoint_freq: 10
checkpoint_at_end: True
keep_checkpoints_num: 5
......
......@@ -2,7 +2,7 @@ flatland-random-sparse-small-tree-fc-ppo:
run: APEX
env: flatland_sparse
stop:
timesteps_total: 50000
timesteps_total: 500
checkpoint_freq: 10
checkpoint_at_end: True
keep_checkpoints_num: 5
......
......@@ -2,7 +2,7 @@ flatland-random-sparse-small-tree-fc-ppo:
run: APEX
env: flatland_sparse
stop:
timesteps_total: 100000000 # 1e8
timesteps_total: 500
checkpoint_freq: 10
checkpoint_at_end: True
keep_checkpoints_num: 5
......
......@@ -17,6 +17,16 @@ echo "===================="
time python ./train.py -f experiments/tests/global_density_obs_apex.yaml
echo "===================="
echo "COMBINED OBS"
echo "LOCAL CONFLICT OBS"
echo "===================="
time python ./train.py -f experiments/tests/local_conflict_obs_apex.yaml
echo "===================="
echo "TREE OBS"
echo "===================="
time python ./train.py -f experiments/tests/tree_obs_apex.yaml
echo "===================="
echo "COMBINED OBS (TREE + LOCAL CONFLICT)"
echo "===================="
time python ./train.py -f experiments/tests/combined_obs_apex.yaml
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment