Commit 00b5d4b5 authored by manuschn's avatar manuschn

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: ...@@ -2,7 +2,7 @@ flatland-sparse-small-density-cnn-apex:
run: APEX run: APEX
env: flatland_sparse env: flatland_sparse
stop: stop:
timesteps_total: 10000 timesteps_total: 500
checkpoint_freq: 10 checkpoint_freq: 10
checkpoint_at_end: True checkpoint_at_end: True
keep_checkpoints_num: 5 keep_checkpoints_num: 5
......
...@@ -2,7 +2,7 @@ flatland-sparse-small-density-cnn-apex: ...@@ -2,7 +2,7 @@ flatland-sparse-small-density-cnn-apex:
run: APEX run: APEX
env: flatland_sparse env: flatland_sparse
stop: stop:
timesteps_total: 10000 timesteps_total: 500
checkpoint_freq: 10 checkpoint_freq: 10
checkpoint_at_end: True checkpoint_at_end: True
keep_checkpoints_num: 5 keep_checkpoints_num: 5
......
...@@ -2,7 +2,7 @@ flatland-sparse-global-conv-ppo: ...@@ -2,7 +2,7 @@ flatland-sparse-global-conv-ppo:
run: PPO run: PPO
env: flatland_sparse env: flatland_sparse
stop: stop:
timesteps_total: 500 # 1e7 timesteps_total: 500
checkpoint_freq: 10 checkpoint_freq: 10
checkpoint_at_end: True checkpoint_at_end: True
keep_checkpoints_num: 5 keep_checkpoints_num: 5
......
...@@ -2,7 +2,7 @@ flatland-random-sparse-small-tree-fc-ppo: ...@@ -2,7 +2,7 @@ flatland-random-sparse-small-tree-fc-ppo:
run: APEX run: APEX
env: flatland_sparse env: flatland_sparse
stop: stop:
timesteps_total: 50000 timesteps_total: 500
checkpoint_freq: 10 checkpoint_freq: 10
checkpoint_at_end: True checkpoint_at_end: True
keep_checkpoints_num: 5 keep_checkpoints_num: 5
......
...@@ -2,7 +2,7 @@ flatland-random-sparse-small-tree-fc-ppo: ...@@ -2,7 +2,7 @@ flatland-random-sparse-small-tree-fc-ppo:
run: APEX run: APEX
env: flatland_sparse env: flatland_sparse
stop: stop:
timesteps_total: 100000000 # 1e8 timesteps_total: 500
checkpoint_freq: 10 checkpoint_freq: 10
checkpoint_at_end: True checkpoint_at_end: True
keep_checkpoints_num: 5 keep_checkpoints_num: 5
......
...@@ -17,6 +17,16 @@ echo "====================" ...@@ -17,6 +17,16 @@ echo "===================="
time python ./train.py -f experiments/tests/global_density_obs_apex.yaml time python ./train.py -f experiments/tests/global_density_obs_apex.yaml
echo "====================" 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 "====================" echo "===================="
time python ./train.py -f experiments/tests/combined_obs_apex.yaml 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