model_kwargs.json

{
  "model_class": "rl.models.GATV2ActorCritic",
  "format": "raw PyTorch state_dict",
  "n_node_features": 16,
  "n_edge_features": 2,
  "proj_out": 64,
  "num_gat_blocks": 4,
  "gat_channels": [
    128,
    128,
    64,
    64
  ],
  "num_heads": [
    8,
    8,
    4,
    4
  ],
  "concat": false,
  "activation": "tanh",
  "attn_dropout": [
    0.0,
    0.0,
    0.0,
    0.0
  ],
  "feat_dropout": [
    0.0,
    0.0,
    0.0,
    0.0
  ],
  "actor_head_dropout": 0.0,
  "critic_head_dropout": 0.0,
  "actor_head_layers": [
    256,
    128,
    64
  ],
  "critic_head_layers": [
    256,
    128,
    64
  ],
  "critic_readout_type": "sum"
}