apex.yaml 880 Bytes
Newer Older
1
2
3
4
flatland-sparse-small-tree-fc-apex:
    run: APEX
    env: flatland_sparse
    stop:
nilabha's avatar
nilabha committed
5
6
        timesteps_total: 15000000  # 1.5e7
    checkpoint_freq: 50
7
    checkpoint_at_end: True
8
    keep_checkpoints_num: 100000000
9
    checkpoint_score_attr: episode_reward_mean
nilabha's avatar
nilabha committed
10
    num_samples: 3
11
    config:
nilabha's avatar
nilabha committed
12
        num_workers: 13
13
14
15
16
17
18
19
20
21
22
23
24
25
        num_envs_per_worker: 5
        num_gpus: 0

        env_config:
            observation: tree
            observation_config:
                max_depth: 2
                shortest_path_max_depth: 30

            generator: sparse_rail_generator
            generator_config: small_v0

            wandb:
nilabha's avatar
nilabha committed
26
27
                project: flatland-paper
                entity: aicrowd
28
29
30
31
32
33
                tags: ["small_v0", "tree_obs", "apex"]

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