combined_obs_apex.yaml 1.14 KB
Newer Older
1 2 3 4
flatland-sparse-small-density-cnn-apex:
    run: APEX
    env: flatland_sparse
    stop:
5
        timesteps_total: 500
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
    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

        env_config:
            observation: combined
            observation_config:
                tree:
                  max_depth: 2
                  shortest_path_max_depth: 30
                localConflict:
                  max_depth: 2
                  shortest_path_max_depth: 30
                  n_local: 5

            generator: sparse_rail_generator
            generator_config: small_v0

            resolve_deadlocks: False
            deadlock_reward: 0
            density_reward_factor: 0

            wandb:
                project: flatland
                entity: masterscrat
                tags: ["small_v0", "tree_and_local_conflict", "apex"] # TODO should be set programmatically

        model:
            fcnet_activation: relu
            fcnet_hiddens: [256, 256]
            vf_share_layers: True