Commit 6f5e851d authored by nilabha's avatar nilabha

Merge branch 'master' of gitlab.aicrowd.com:flatland/neurips2020-flatland-baselines

parents 5243dd6a 4fee3b4e
......@@ -25,8 +25,8 @@ flatland-random-sparse-small-tree-fc-apex-il:
generator_config: small_v0
wandb:
project: nilabha2007
entity: neurips2020-flatland-baselines
project: flatland
entity: masterscrat
tags: ["small_v0", "tree_obs", "APEX_DQfD"] # TODO should be set programmatically
model:
......@@ -36,6 +36,7 @@ flatland-random-sparse-small-tree-fc-apex-il:
vf_share_layers: True # False
custom_options:
input_files: /tmp/flatland-out
expert_size: 1
loss: dqfd # ce (cross entropy), kl (kl divergence)
lambda1: 1
lambda2: 1
......