model_configs.json

{
  "gin_gru_d1_w5": {
    "in_dim": 135,
    "num_action": 7,
    "gin_hidden": 32,
    "gru_hidden": 64,
    "gin_layers": 2,
    "gru_layers": 1,
    "dropout": 0.4,
    "window": 5,
    "action_mask": [
      0,
      2,
      4,
      7,
      9,
      12,
      14
    ]
  },
  "gin_gru_d2_w5": {
    "in_dim": 135,
    "num_action": 7,
    "gin_hidden": 64,
    "gru_hidden": 128,
    "gin_layers": 3,
    "gru_layers": 1,
    "dropout": 0.3,
    "window": 5,
    "action_mask": [
      0,
      2,
      4,
      7,
      9,
      12,
      14
    ]
  }
}