Commit bbebe763 authored by manueth's avatar manueth

test for global density obs

parent fad53fb0
flatland-sparse-small-density-cnn-apex:
run: APEX
env: flatland_sparse
stop:
timesteps_total: 10000
checkpoint_freq: 10
checkpoint_at_end: True
keep_checkpoints_num: 5
checkpoint_score_attr: episode_reward_mean
config:
num_workers: 2
num_envs_per_worker: 5
num_gpus: 0
hiddens: []
dueling: False
env_config:
observation: density
observation_config:
width: 25
height: 25
max_t: 10
encoding: exp_decay
generator: sparse_rail_generator
generator_config: small_v0
wandb:
project: flatland
entity: masterscrat
tags: ["small_v0", "density_obs", "apex"] # TODO should be set programmatically
model:
custom_model: global_dens_obs_model
custom_options:
architecture: impala
architecture_options:
residual_layers: [[16, 2], [32, 4]]
......@@ -6,4 +6,9 @@ time python ./trainImitate.py -f experiments/tests/MARWIL.yaml
echo "===================="
echo "GLOBAL OBS"
echo "===================="
time python ./train.py -f experiments/tests/global_obs.yaml
\ No newline at end of file
time python ./train.py -f experiments/tests/global_obs.yaml
echo "===================="
echo "GLOBAL DENSITY OBS"
echo "===================="
time python ./train.py -f experiments/tests/global_density_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