Publish AMARO2 model card; checkpoints on the way
Browse files- README.md +3 -18
- SHA256SUMS +1 -1
README.md
CHANGED
|
@@ -44,9 +44,6 @@ The checkpoint files and `v2.0` release tag are not available yet.
|
|
| 44 |
| --- | --- | --- |
|
| 45 |
| `v2.0` | `amaro_v2_sm.ckpt`, `amaro_v2_p.ckpt`, `amaro_v2.ckpt` | Extended no-hydrogen (NOH) solute mapping |
|
| 46 |
|
| 47 |
-
Once the checkpoints are published, use a release tag or full commit SHA for
|
| 48 |
-
reproducible downloads. `main` follows the latest model card and can change
|
| 49 |
-
between releases.
|
| 50 |
|
| 51 |
## AMARO v2 training stages
|
| 52 |
|
|
@@ -59,14 +56,6 @@ sequence. Each stage continues training from the preceding stage:
|
|
| 59 |
| 2: proteins | `amaro_v2_p.ckpt` | Protein conformations from mdCATH, with molecular-mechanics force labels | Intermediate checkpoint after protein training |
|
| 60 |
| 3: protein–ligand complexes | `amaro_v2.ckpt` | Acellera HoloSet protein–ligand conformations, with molecular-mechanics force labels | Final AMARO v2.0 model in the preprint |
|
| 61 |
|
| 62 |
-
Use `amaro_v2.ckpt` to reproduce the final model evaluated for biomolecular
|
| 63 |
-
complex dynamics. Use `amaro_v2_sm.ckpt` for the preprint's small-molecule-stage
|
| 64 |
-
comparisons, and `amaro_v2_p.ckpt` for comparisons with the intermediate protein
|
| 65 |
-
stage. The final model does not uniformly improve every property: the preprint
|
| 66 |
-
reports lower small-molecule force errors and closer local-geometry distributions
|
| 67 |
-
for the SM checkpoint, while the final model is more robust at the tested MD
|
| 68 |
-
time steps. No separate validation claim is made here for the protein checkpoint.
|
| 69 |
-
|
| 70 |
### Inputs and outputs
|
| 71 |
|
| 72 |
AMARO v2 uses NOH species IDs, which encode retained elements and, where
|
|
@@ -111,11 +100,9 @@ checkpoint = hf_hub_download(
|
|
| 111 |
)
|
| 112 |
```
|
| 113 |
|
| 114 |
-
Select another v2 variant by changing `filename`.
|
| 115 |
-
this model card and its figure. The checkpoint release will also include
|
| 116 |
-
checksums for the three checkpoint files at the same revision.
|
| 117 |
|
| 118 |
-
To download all three training stages:
|
| 119 |
|
| 120 |
```bash
|
| 121 |
hf download compsciencelab/AMARO2 \
|
|
@@ -125,9 +112,7 @@ hf download compsciencelab/AMARO2 \
|
|
| 125 |
|
| 126 |
## License
|
| 127 |
|
| 128 |
-
AMARO v2 checkpoint licensing
|
| 129 |
-
does not specify a checkpoint license. Licensing of the original v1 release
|
| 130 |
-
does not establish licensing for these v2 checkpoints.
|
| 131 |
|
| 132 |
## Citation
|
| 133 |
|
|
|
|
| 44 |
| --- | --- | --- |
|
| 45 |
| `v2.0` | `amaro_v2_sm.ckpt`, `amaro_v2_p.ckpt`, `amaro_v2.ckpt` | Extended no-hydrogen (NOH) solute mapping |
|
| 46 |
|
|
|
|
|
|
|
|
|
|
| 47 |
|
| 48 |
## AMARO v2 training stages
|
| 49 |
|
|
|
|
| 56 |
| 2: proteins | `amaro_v2_p.ckpt` | Protein conformations from mdCATH, with molecular-mechanics force labels | Intermediate checkpoint after protein training |
|
| 57 |
| 3: protein–ligand complexes | `amaro_v2.ckpt` | Acellera HoloSet protein–ligand conformations, with molecular-mechanics force labels | Final AMARO v2.0 model in the preprint |
|
| 58 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
### Inputs and outputs
|
| 60 |
|
| 61 |
AMARO v2 uses NOH species IDs, which encode retained elements and, where
|
|
|
|
| 100 |
)
|
| 101 |
```
|
| 102 |
|
| 103 |
+
Select another v2 variant by changing `filename`.
|
|
|
|
|
|
|
| 104 |
|
| 105 |
+
To download all three ckpts training stages:
|
| 106 |
|
| 107 |
```bash
|
| 108 |
hf download compsciencelab/AMARO2 \
|
|
|
|
| 112 |
|
| 113 |
## License
|
| 114 |
|
| 115 |
+
AMARO v2 checkpoint licensing to define.
|
|
|
|
|
|
|
| 116 |
|
| 117 |
## Citation
|
| 118 |
|
SHA256SUMS
CHANGED
|
@@ -1,2 +1,2 @@
|
|
| 1 |
-
|
| 2 |
1851f74f78f57a5e644036278c9f75e8509238edabb53a5e280c9f0e8c70f7a5 assets/social-preview-fig-amaro2.webp
|
|
|
|
| 1 |
+
8a4aa695e1fef8af2e3f9c77107efa4f988241e53973fd3fc2c44004879ba4ef README.md
|
| 2 |
1851f74f78f57a5e644036278c9f75e8509238edabb53a5e280c9f0e8c70f7a5 assets/social-preview-fig-amaro2.webp
|