env_config:
  lib: expground.envs.gathering
  env_id: gathering_small
  params:
    n_agents: 2
    map_name: small

algorithm:
  name: DQN
  policy: dqn.DQN
  trainer: dqn.DQNTrainer
  loss: dqn.loss.DQNLoss

model_config:
  critic:
    network: mlp
    layers:
      - units: 256
        activation: ReLU
      - units: 256
        activation: ReLU
      - units: 256
        activation: ReLU
      - units: 256
        activation: ReLU
    output:
      activation: False

custom_config:
  use_cuda: False
  preprocess_mode: stack
  # num_actor: fragment_length // max_step
  use_vector_env: False # for eval