{
  "model": "sage",
  "feature": "bert",
  "in_dim": 768,
  "hidden": 128,
  "layers": 2,
  "heads": 4,
  "dropout": 0.5,
  "edge_drop": 0.2,
  "admm_p": 2.5,
  "admm_lambda": 1.0,
  "admm_iterations": 2,
  "feature_l1": 0.001,
  "epochs": 200,
  "batch_size": 128,
  "lr": 0.001,
  "weight_decay": 0.0001
}