config.json

{
  "hidden_dims": [
    512,
    512
  ],
  "input_features_dim": 89,
  "num_blocks": 16,
  "num_heads": 4,
  "num_hops": 8,
  "output_features_dim": 83,
  "scale_factor": 1.0,
  "sparse": true,
  "splits": 4,
  "use_edges_features": false
}