Skip to content

Commit 4631235

Browse files
add information on changing checkpoints (#39)
Co-authored-by: Han Yang <[email protected]>
1 parent 50a1ba2 commit 4631235

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

README.md

+7
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,13 @@ print(f"Stress[0][0] (eV/A^3) = {si.get_stress(voigt=False)[0][0]}")
8686
print(f"Stress[0][0] (GPa) = {si.get_stress(voigt=False)[0][0] / GPa}")
8787
```
8888

89+
In this release, we provide two checkpoints: `MatterSim-v1.0.0-1M.pth` and `MatterSim-v1.0.0-5M.pth`. By default, the `1M` version is loaded.
90+
To switch to the `5M` version, manually set the `load_path` of `MatterSimCalculator` as shown below:
91+
92+
```python
93+
MatterSimCalculator(load_path="MatterSim-v1.0.0-5M.pth", device=device)
94+
```
95+
8996

9097
## Reference
9198
We kindly request that users of MatterSim version 1.0.0 cite our preprint available on arXiv:

0 commit comments

Comments
 (0)