config.json

{
  "model_type": "fragnet",
  "n_classes": 1,
  "atom_features": 167,
  "frag_features": 167,
  "edge_features": 17,
  "num_layer": 4,
  "num_heads": 4,
  "drop_ratio": 0.15,
  "h1": 256,
  "h2": 256,
  "h3": 256,
  "h4": 256,
  "act": "celu",
  "emb_dim": 128,
  "fthead": "FTHead2",
  "task_type": "regression"
}