Commit b88ae4a6 authored by MasterScrat's avatar MasterScrat

Random agent, alternative small stoch env

parent 8b082b51
width: 25
height: 25
number_of_agents: 5
max_num_cities: 4
grid_mode: False
max_rails_between_cities: 2
max_rails_in_city: 3
malfunction_rate: 200
malfunction_min_duration: 5
malfunction_max_duration: 20
seed: 0
regenerate_rail_on_reset: True
regenerate_schedule_on_reset: True
\ No newline at end of file
flatland-random-sparse-small-tree-fc-ppo:
run: PPO
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:
lr: 0.0 # random agent!
num_workers: 2
num_envs_per_worker: 2
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:
project: flatland
entity: masterscrat
tags: ["small_v0", "tree_obs", "random"] # TODO should be set programmatically
\ No newline at end of file
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