File size: 264 Bytes
782c7ef
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
{
  "model_type": "ProtNHF",
  "architectural_version": 1,
  "dt": 0.05,
  "niter": 4,
  "hidden_dims": 128,
  "std": 0.7,
  "integrator": "leapfrog",
  "n_types": 20,
  "energy": {
    "d_model": 320,
    "ff_dim": 1280,
    "n_heads": 20,
    "n_layers": 6
  }
}