AntonioMirarchi commited on
Commit
0dcc554
·
verified ·
1 Parent(s): 72d487f

Publish AMARO2 model card; checkpoints on the way

Browse files
Files changed (2) hide show
  1. README.md +3 -18
  2. 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`. For now, `SHA256SUMS` covers
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 is pending author confirmation. The preprint
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
- cffa64c268873c1994abfa5c9609a6e3a7d3ad5977fe266a4255da03183bd754 README.md
2
  1851f74f78f57a5e644036278c9f75e8509238edabb53a5e280c9f0e8c70f7a5 assets/social-preview-fig-amaro2.webp
 
1
+ 8a4aa695e1fef8af2e3f9c77107efa4f988241e53973fd3fc2c44004879ba4ef README.md
2
  1851f74f78f57a5e644036278c9f75e8509238edabb53a5e280c9f0e8c70f7a5 assets/social-preview-fig-amaro2.webp