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

experiments/tests/combined_obs_apex.yaml

@@ -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

experiments/tests/global_density_obs_apex.yaml

@@ -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

experiments/tests/global_obs_ppo.yaml

@@ -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

experiments/tests/conflict_obs_apex.yaml → experiments/tests/local_conflict_obs_apex.yaml

@@ -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

experiments/tests/tree_obs_apex.yaml

@@ -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

test.sh

@@ -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