apex.yaml 916 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10
flatland-random-sparse-small-tree-fc-ppo:
    run: APEX
    env: flatland_sparse
    stop:
        timesteps_total: 100000000  # 1e8
    checkpoint_freq: 10
    checkpoint_at_end: True
    keep_checkpoints_num: 5
    checkpoint_score_attr: episode_reward_mean
    config:
11
        num_workers: 19
12 13 14 15 16 17 18 19 20 21
        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
22
            generator_config: medium_stoch_v1
23 24 25 26

            wandb:
                project: flatland
                entity: masterscrat
27
                tags: ["medium_stoch_v1", "tree_obs", "apex"] # TODO should be set programmatically
28 29 30 31 32

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