Skip to content

Commit d060441

Browse files
authored
Update README.md
1 parent 171a3fd commit d060441

File tree

1 file changed

+10
-18
lines changed

1 file changed

+10
-18
lines changed

README.md

Lines changed: 10 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -46,17 +46,11 @@ calc = SevenNetCalculator('7net-mf-ompa', modal='mpa') # Use modal='omat24' for
4646
4747
When using the command-line interface of SevenNet, include the `--modal mpa` or `--modal omat24` option to select the desired modality.
4848

49-
5049
#### **Matbench Discovery**
5150
| CPS | F1 | $\kappa_{\mathrm{SRME}}$ | RMSD |
5251
|:---:|:---:|:---:|:---:|
5352
|**0.845**|**0.901**|0.317| **0.064** |
5453

55-
**Maximum Atoms Supported**
56-
57-
~ **1,000 atoms** (Bulk / ASE `SevenNetCalculator` / single 24GB VRAM GPU) \
58-
※ This number is approximate and may vary depending on your specific GPU memory (VRAM) capacity and execution environment (e.g., calculation with LAMMPS, multi-GPU...).
59-
6054
[Detailed instructions for multi-fidelity learning](https://github.com/MDIL-SNU/SevenNet/blob/main/sevenn/pretrained_potentials/SevenNet_MF_0/README.md)
6155

6256
[Download link for fully detailed checkpoint](https://figshare.com/articles/software/7net_MF_ompa/28590722?file=53029859)
@@ -72,10 +66,6 @@ When using the command-line interface of SevenNet, include the `--modal mpa` or
7266
#### **Matbench Discovery**
7367
* $\kappa_{\mathrm{SRME}}$: **0.221**
7468

75-
#### **Maximum Atoms Supported**
76-
77-
~ **1,600 atoms** (Bulk / ASE `SevenNetCalculator` / single 24GB VRAM GPU)
78-
7969
---
8070
### **SevenNet-l3i5 (12Dec2024)**
8171
> Model keywords: `7net-l3i5` | `SevenNet-l3i5`
@@ -87,10 +77,6 @@ This model increases the maximum spherical harmonic degree ($l_{\mathrm{max}}$)
8777
|:---:|:---:|:---:|:---:|
8878
|0.714 |0.760|0.550|0.085|
8979

90-
#### **Maximum Atoms Supported**
91-
92-
~ **2,800 atoms** (Bulk / ASE `SevenNetCalculator` / single 24GB VRAM GPU)
93-
9480
---
9581

9682
### **SevenNet-0 (11Jul2024)**
@@ -104,10 +90,6 @@ For more information, click [here](sevenn/pretrained_potentials/SevenNet_0__11Ju
10490
|:---:|:---:|
10591
|0.67|0.767|
10692

107-
#### **Maximum Atoms Supported**
108-
109-
~ **6,500 atoms** (Bulk / ASE `SevenNetCalculator` / single 24GB VRAM GPU)
110-
11193
---
11294
You can find our legacy models in [pretrained_potentials](./sevenn/pretrained_potentials).
11395

@@ -166,6 +148,16 @@ If you encounter the error `CUDA is not installed or nvcc is not available`, ple
166148

167149
Various pretrained SevenNet models can be accessed by setting the model variable to predefined keywords like `7net-mf-ompa`, `7net-omat`, `7net-l3i5`, and `7net-0`.
168150

151+
The following table provides **approximate** maximum atom counts of **A100 GPU (80GB)** in a bulk system.
152+
| Model | Max atoms |
153+
|:---:|:---:|
154+
|7net-0|~ 21,500|
155+
|7net-l3i5|~ 9,300|
156+
|7net-omat|~ 5,300|
157+
|7net-mf-ompa|~ 3,300|
158+
159+
Note that this value depends on the target system. The limitation can be overcome by leveraging multi-GPUs with LAMMPS.
160+
169161
Additionally, user-trained models can be applied with the ASE calculator. In this case, the `model` parameter should be set to the checkpoint path from training.
170162

171163
> [!TIP]

0 commit comments

Comments
 (0)