| { | |
| "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 | |
| } | |
| } |
| { | |
| "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 | |
| } | |
| } |