fairchem by FAIR Chemistry
+
+fairchem by FAIR Chemistry
+#### FAIR-Chem overview
+
`fairchem` is the [FAIR](https://ai.meta.com/research/) Chemistry's centralized repository of all its data, models, demos, and application efforts
for materials science and quantum chemistry. Collaborative projects that contribute or use the models and approaches in
this repo:
@@ -22,7 +25,7 @@ We re-organized and rebranded the repository in 2024 (previously the `fairchem`
general usability of these models beyond catalysis, including things like direct air capture.
```
-### Datasets in `fairchem`:
+#### Datasets in `fairchem`:
`fairchem` provides training and evaluation code for tasks and models that take arbitrary
chemical structures as input to predict energies / forces / positions / stresses,
and can be used as a base scaffold for research projects. For an overview of
@@ -33,7 +36,7 @@ tasks, data, and metrics, please read the documentations and respective papers:
- [OC20Dense](core/datasets/oc20dense)
- [OC20NEB](core/datasets/oc20neb)
-### Projects and models built on `fairchem`:
+#### Projects and models built on `fairchem`:
- SchNet [[`arXiv`](https://arxiv.org/abs/1706.08566)] [[`code`](https://github.com/FAIR-Chem/fairchem/blob/main/src/fairchem/core/models/schnet.py)]
- DimeNet++ [[`arXiv`](https://arxiv.org/abs/2011.14115)] [[`code`](https://github.com/FAIR-Chem/fairchem/blob/main/src/fairchem/core/models/dimenet_plus_plus.py)]
@@ -53,7 +56,7 @@ Older model implementations that are no longer supported:
- SpinConv [[`arXiv`](https://arxiv.org/abs/2106.09575)] [[`code`](https://github.com/FAIR-Chem/fairchem/blob/e7a8745eb307e8a681a1aa9d30c36e8c41e9457e/ocpmodels/models/spinconv.py)]
- ForceNet [[`arXiv`](https://arxiv.org/abs/2103.01436)] [[`code`](https://github.com/FAIR-Chem/fairchem/blob/e7a8745eb307e8a681a1aa9d30c36e8c41e9457e/ocpmodels/models/forcenet.py)]
-## Discussion
+### Discussion
For all non-codebase related questions and to keep up-to-date with the latest OCP
announcements, please join the [discussion board](https://discuss.opencatalystproject.org/).
@@ -61,7 +64,7 @@ announcements, please join the [discussion board](https://discuss.opencatalystpr
All code-related questions and issues should be posted directly on our
[issues page](https://github.com/FAIR-Chem/fairchem/issues).
-## Acknowledgements
+### Acknowledgements
- This codebase was initially forked from [CGCNN](https://github.com/txie-93/cgcnn)
by [Tian Xie](http://txie.me), but has undergone significant changes since.
@@ -70,11 +73,11 @@ by [Tian Xie](http://txie.me), but has undergone significant changes since.
- It was then developed as the OCP repo, and includes many contributions from the community and collaborators.
- Much of the documentation was developed for various papers or as part of a comprehensive tutorial for the 2023 ACS Fall Chemistry conference.
-## License
+### License
`fairchem` is released under the [MIT](https://github.com/FAIR-Chem/fairchem/blob/main/LICENSE.md) license.
-## Citing `fairchem`
+### Citing `fairchem`
If you use this codebase in your work, please consider citing:
diff --git a/_static/logo.png b/_static/logo.png
deleted file mode 100644
index ffb62c1197..0000000000
Binary files a/_static/logo.png and /dev/null differ
diff --git a/autoapi/adsorbml/2023_neurips_challenge/challenge_eval/index.html b/autoapi/adsorbml/2023_neurips_challenge/challenge_eval/index.html
index 1abbaf8cf8..14446143cb 100644
--- a/autoapi/adsorbml/2023_neurips_challenge/challenge_eval/index.html
+++ b/autoapi/adsorbml/2023_neurips_challenge/challenge_eval/index.html
@@ -8,7 +8,7 @@
-
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
-
+
from fairchem.core.models.model_registry import model_name_to_local_file
+from fairchem.core.models.model_registry import model_name_to_local_file
checkpoint_path = model_name_to_local_file('GemNet-OC-S2EFS-OC20+OC22', local_cache='/tmp/ocp_checkpoints/')
@@ -660,7 +654,7 @@ Fine tuning a model
-import json
+import json
import numpy as np
import matplotlib.pyplot as plt
from ase import Atoms
@@ -675,66 +669,11 @@ Fine tuning a model
-(['SnO2', 'IrO2', 'RuO2', 'TiO2', 'VO2'],
- ['rutile', 'pyrite', 'columbite', 'brookite', 'fluorite', 'anatase'])
-
-
-
A specific calculation has a lot of details in it. You can use this to recreate the calculations. Shortly we will use these to compare the DFT results to OCP.
-d['TiO2']['rutile']['PBE']['EOS']['calculations'][0]
-
-
-
-
-{'incar': {'doc': 'INCAR parameters',
- 'prec': 'Normal',
- 'isif': 4,
- 'nbands': 20,
- 'ibrion': 2,
- 'gga': 'PE',
- 'encut': 520.0,
- 'ismear': 0,
- 'sigma': 0.001,
- 'nsw': 50},
- 'doc': 'JSON representation of a VASP calculation.\n\nenergy is in eV\nforces are in eV/\\AA\nstress is in GPa (sxx, syy, szz, syz, sxz, sxy)\nmagnetic moments are in Bohr-magneton\nThe density of states is reported with E_f at 0 eV.\nVolume is reported in \\AA^3\nCoordinates and cell parameters are reported in \\AA\n\nIf atom-projected dos are included they are in the form:\n{ados:{energy:data, {atom index: {orbital : dos}}}\n',
- 'potcar': [['O',
- 'potpaw/O/POTCAR',
- '0cf2ce56049ca395c567026b700ed66c94a85161'],
- ['Ti', 'potpaw/Ti/POTCAR', '51f7f05982d6b4052becc160375a8b8b670177a7']],
- 'input': {'kpts': [6, 6, 6],
- 'reciprocal': False,
- 'xc': 'LDA',
- 'kpts_nintersections': None,
- 'setups': None,
- 'txt': '-',
- 'gamma': False},
- 'atoms': {'cell': [[4.3789762519649225, 0.0, 0.0],
- [0.0, 4.3789762519649225, 0.0],
- [0.0, 0.0, 2.864091775985314]],
- 'symbols': ['Ti', 'Ti', 'O', 'O', 'O', 'O'],
- 'tags': [0, 0, 0, 0, 0, 0],
- 'pbc': [True, True, True],
- 'positions': [[0.0, 0.0, 0.0],
- [2.1894881259824612, 2.1894881259824612, 1.432045887992657],
- [1.3181554154438013, 1.3181554154438013, 0.0],
- [3.0608208365211214, 3.0608208365211214, 0.0],
- [3.5076435414262623, 0.87133271053866, 1.432045887992657],
- [0.87133271053866, 3.5076435414262623, 1.432045887992657]]},
- 'data': {'stress': [496.18519999, 496.18519999, 502.82679392, 0.0, 0.0, 0.0],
- 'doc': 'Data from the output of the calculation',
- 'volume': 54.92019999999996,
- 'total_energy': -56.230672,
- 'forces': [[0.0, 0.0, 0.0],
- [0.0, 0.0, 0.0],
- [-0.001264, -0.001264, 0.0],
- [0.001264, 0.001264, 0.0],
- [-0.001264, 0.001264, 0.0],
- [0.001264, -0.001264, 0.0]],
- 'fermi_level': 3.153}}
+d['TiO2']['rutile']['PBE']['EOS']['calculations'][0]
@@ -742,7 +681,7 @@ Fine tuning a model
-c = d['TiO2']['rutile']['PBE']['EOS']['calculations'][0]
+c = d['TiO2']['rutile']['PBE']['EOS']['calculations'][0]
atoms = Atoms(symbols=c['atoms']['symbols'],
positions=c['atoms']['positions'],
cell=c['atoms']['cell'],
@@ -752,46 +691,20 @@ Fine tuning a model
-(Atoms(symbols='Ti2O4', pbc=True, cell=[4.3789762519649225, 4.3789762519649225, 2.864091775985314], tags=...),
- -56.230672,
- [[0.0, 0.0, 0.0],
- [0.0, 0.0, 0.0],
- [-0.001264, -0.001264, 0.0],
- [0.001264, 0.001264, 0.0],
- [-0.001264, 0.001264, 0.0],
- [0.001264, -0.001264, 0.0]])
-
-
-
Next, we will create an OCP calculator that we can use to get predictions from.
-from fairchem.core.common.relaxation.ase_utils import OCPCalculator
+from fairchem.core.common.relaxation.ase_utils import OCPCalculator
calc = OCPCalculator(checkpoint_path=checkpoint_path, trainer='forces', cpu=False)
-
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Unrecognized arguments: ['symmetric_edge_symmetrization']
-
-
-WARNING:root:Using `weight_decay` from `optim` instead of `optim.optimizer_params`.Please update your config to use `optim.optimizer_params.weight_decay`.`optim.weight_decay` will soon be deprecated.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-
Now, we loop through each structure and accumulate the OCP predictions. Then, we plot the parity results.
-import time
+import time
t0 = time.time()
eos_data = {}
@@ -825,31 +738,20 @@ Fine tuning a model
-Elapsed time 67.2 seconds.
-
-
-
-
-mae = np.mean(np.abs(np.array(dft) - np.array(ocp)))
+mae = np.mean(np.abs(np.array(dft) - np.array(ocp)))
print(f'MAE = {mae:1.3f} eV/atom')
-
-MAE = 1.669 eV/atom
-
-
-
The MAE is somewhat high compared to the reported value of 0.458 eV. That is not too surprising; although OC22 was trained on oxides, it was not trained on all of these structures. It is also evident on inspection that the main issues are all the VO2 structures, and these skew the MAE. Next we look at the EOS for each material. Here you can see offsets, and qualitative issues in the shapes. The offset is largely a PBE/RPBE difference.
Some notable issues are with fluorite structures. We show one here for VO2.
-oxide, polymorph = 'VO2', 'fluorite'
+oxide, polymorph = 'VO2', 'fluorite'
V, D, O = eos_data[oxide][polymorph]
plt.plot(V, D, label='dft')
@@ -859,9 +761,6 @@ Fine tuning a model
-
-
We can improve this by fine-tuning the model.
@@ -877,10 +776,10 @@ Fine tuning the checkpoint
-from ase.db import connect
+from ase.db import connect
from ase.calculators.singlepoint import SinglePointCalculator
-! rm -fr oxides.db # start clean when you run this cell
+! rm -fr oxides.db # start clean when you run this cell
db = connect('oxides.db')
@@ -888,7 +787,7 @@ Fine tuning the checkpoint
-for oxide in oxides:
+for oxide in oxides:
for polymorph in polymorphs:
for c in d[oxide][polymorph]['PBE']['EOS']['calculations']:
atoms = Atoms(symbols=c['atoms']['symbols'],
@@ -908,33 +807,7 @@ Fine tuning the checkpoint
-! ase db oxides.db
-
-
-
-
-id|age|user |formula|calculator| energy|natoms| fmax|pbc| volume|charge| mass
- 1| 1s|runner|Sn2O4 |unknown |-41.359| 6|0.045|TTT| 64.258| 0.000|301.416
- 2| 1s|runner|Sn2O4 |unknown |-41.853| 6|0.025|TTT| 66.526| 0.000|301.416
- 3| 1s|runner|Sn2O4 |unknown |-42.199| 6|0.010|TTT| 68.794| 0.000|301.416
- 4| 1s|runner|Sn2O4 |unknown |-42.419| 6|0.006|TTT| 71.062| 0.000|301.416
- 5| 1s|runner|Sn2O4 |unknown |-42.534| 6|0.011|TTT| 73.330| 0.000|301.416
- 6| 1s|runner|Sn2O4 |unknown |-42.562| 6|0.029|TTT| 75.598| 0.000|301.416
- 7| 1s|runner|Sn2O4 |unknown |-42.518| 6|0.033|TTT| 77.866| 0.000|301.416
- 8| 1s|runner|Sn2O4 |unknown |-42.415| 6|0.010|TTT| 80.134| 0.000|301.416
- 9| 1s|runner|Sn2O4 |unknown |-42.266| 6|0.006|TTT| 82.402| 0.000|301.416
-10| 1s|runner|Sn2O4 |unknown |-42.083| 6|0.017|TTT| 84.670| 0.000|301.416
-11| 1s|runner|Sn4O8 |unknown |-81.424| 12|0.012|TTT|117.473| 0.000|602.832
-12| 1s|runner|Sn4O8 |unknown |-82.437| 12|0.005|TTT|121.620| 0.000|602.832
-13| 1s|runner|Sn4O8 |unknown |-83.147| 12|0.015|TTT|125.766| 0.000|602.832
-14| 1s|runner|Sn4O8 |unknown |-83.599| 12|0.047|TTT|129.912| 0.000|602.832
-15| 1s|runner|Sn4O8 |unknown |-83.831| 12|0.081|TTT|134.058| 0.000|602.832
-16| 1s|runner|Sn4O8 |unknown |-83.898| 12|0.001|TTT|138.204| 0.000|602.832
-17| 1s|runner|Sn4O8 |unknown |-83.805| 12|0.001|TTT|142.350| 0.000|602.832
-18| 1s|runner|Sn4O8 |unknown |-83.586| 12|0.002|TTT|146.496| 0.000|602.832
-19| 1s|runner|Sn4O8 |unknown |-83.262| 12|0.002|TTT|150.642| 0.000|602.832
-20| 1s|runner|Sn4O8 |unknown |-82.851| 12|0.013|TTT|154.788| 0.000|602.832
-Rows: 295 (showing first 20)
+! ase db oxides.db
@@ -947,21 +820,14 @@ Make the train, test, val splitsfairchem.core.common.tutorial_utils to streamline this process.
-from fairchem.core.common.tutorial_utils import train_test_val_split
-! rm -fr train.db test.db val.db
+from fairchem.core.common.tutorial_utils import train_test_val_split
+! rm -fr train.db test.db val.db
train, test, val = train_test_val_split('oxides.db')
train, test, val
-
-(PosixPath('/home/runner/work/fairchem/fairchem/docs/core/fine-tuning/train.db'),
- PosixPath('/home/runner/work/fairchem/fairchem/docs/core/fine-tuning/test.db'),
- PosixPath('/home/runner/work/fairchem/fairchem/docs/core/fine-tuning/val.db'))
-
-
-
@@ -969,7 +835,7 @@ Setting up the configuration yaml file
-from fairchem.core.common.tutorial_utils import generate_yml_config
+from fairchem.core.common.tutorial_utils import generate_yml_config
yml = generate_yml_config(checkpoint_path, 'config.yml',
delete=['slurm', 'cmd', 'logger', 'task', 'model_attributes',
@@ -999,159 +865,10 @@ Setting up the configuration yaml file
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Unrecognized arguments: ['symmetric_edge_symmetrization']
-
-
-WARNING:root:Using `weight_decay` from `optim` instead of `optim.optimizer_params`.Please update your config to use `optim.optimizer_params.weight_decay`.`optim.weight_decay` will soon be deprecated.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-PosixPath('/home/runner/work/fairchem/fairchem/docs/core/fine-tuning/config.yml')
-
-
-
-! cat config.yml
-
-
-
-
-amp: true
-checkpoint: /tmp/ocp_checkpoints/gnoc_oc22_oc20_all_s2ef.pt
-dataset:
- test:
- a2g_args:
- r_energy: false
- r_forces: false
- src: test.db
- train:
- a2g_args:
- r_energy: true
- r_forces: true
- src: train.db
- val:
- a2g_args:
- r_energy: true
- r_forces: true
- src: val.db
-evaluation_metrics:
- metrics:
- energy:
- - mae
- forces:
- - forcesx_mae
- - forcesy_mae
- - forcesz_mae
- - mae
- - cosine_similarity
- - magnitude_error
- misc:
- - energy_forces_within_threshold
- primary_metric: forces_mae
-gpus: 1
-logger: tensorboard
-loss_fns:
-- energy:
- coefficient: 1
- fn: mae
-- forces:
- coefficient: 1
- fn: l2mae
-model:
- activation: silu
- atom_edge_interaction: true
- atom_interaction: true
- cbf:
- name: spherical_harmonics
- cutoff: 12.0
- cutoff_aeaint: 12.0
- cutoff_aint: 12.0
- cutoff_qint: 12.0
- direct_forces: true
- edge_atom_interaction: true
- emb_size_aint_in: 64
- emb_size_aint_out: 64
- emb_size_atom: 256
- emb_size_cbf: 16
- emb_size_edge: 512
- emb_size_quad_in: 32
- emb_size_quad_out: 32
- emb_size_rbf: 16
- emb_size_sbf: 32
- emb_size_trip_in: 64
- emb_size_trip_out: 64
- envelope:
- exponent: 5
- name: polynomial
- extensive: true
- forces_coupled: false
- max_neighbors: 30
- max_neighbors_aeaint: 20
- max_neighbors_aint: 1000
- max_neighbors_qint: 8
- name: gemnet_oc
- num_after_skip: 2
- num_atom: 3
- num_atom_emb_layers: 2
- num_before_skip: 2
- num_blocks: 4
- num_concat: 1
- num_global_out_layers: 2
- num_output_afteratom: 3
- num_radial: 128
- num_spherical: 7
- otf_graph: true
- output_init: HeOrthogonal
- qint_tags:
- - 1
- - 2
- quad_interaction: true
- rbf:
- name: gaussian
- regress_forces: true
- sbf:
- name: legendre_outer
- symmetric_edge_symmetrization: false
-noddp: false
-optim:
- batch_size: 4
- clip_grad_norm: 10
- ema_decay: 0.999
- energy_coefficient: 1
- eval_batch_size: 16
- eval_every: 10
- factor: 0.8
- force_coefficient: 1
- load_balancing: atoms
- loss_energy: mae
- lr_initial: 0.0005
- max_epochs: 1
- mode: min
- num_workers: 2
- optimizer: AdamW
- optimizer_params:
- amsgrad: true
- patience: 3
- scheduler: ReduceLROnPlateau
- weight_decay: 0
-outputs:
- energy:
- level: system
- forces:
- eval_on_free_atoms: true
- level: atom
- train_on_free_atoms: true
-task:
- dataset: ase_db
-trainer: ocp
+! cat config.yml
@@ -1170,70 +887,43 @@ Running the training jobtail -f train.txt
-You can also visit it in a browser at train.txt. You have to periodically refresh the view to see updates though.
+You can also visit it in a browser at train.txt. You have to periodically refresh the view to see updates though.
This can take up to 30 minutes for 80 epochs, so we only do a few here to see what happens.
-
-import time
+
+import time
from fairchem.core.common.tutorial_utils import fairchem_main
t0 = time.time()
-! python {fairchem_main()} --mode train --config-yml {yml} --checkpoint {checkpoint_path} --run-dir fine-tuning --identifier ft-oxides --amp > train.txt 2>&1
+! python {fairchem_main()} --mode train --config-yml {yml} --checkpoint {checkpoint_path} --run-dir fine-tuning --identifier ft-oxides --amp > train.txt 2>&1
print(f'Elapsed time = {time.time() - t0:1.1f} seconds')
-
-
-Show code cell output
-
-
-
-Elapsed time = 204.9 seconds
-
-
-
-
-Now since we have a file, we can find the training results in it. See train.txt. At the top, the config is printed, so we can get the checkpoint directory. I use shell commands and Python to get the line, split and strip it here.
+Now since we have a file, we can find the training results in it. See train.txt. At the top, the config is printed, so we can get the checkpoint directory. I use shell commands and Python to get the line, split and strip it here.
-cpline = !grep "checkpoint_dir:" train.txt
+cpline = !grep "checkpoint_dir:" train.txt
cpdir = cpline[0].split(':')[-1].strip()
cpdir
-
-'fine-tuning/checkpoints/2024-05-17-16-40-32-ft-oxides'
-
-
-
There will be two files in there: checkpoint.pt and best_checkpoint.pt.
The best_checkpoint.pt is the one that performs best on the validation dataset. The checkpoint.pt is the most recently saved one. Probably it has the lowest loss in training, but this could be an indication of overfitting. You have to use some judgement in determining which one to use, and if it is sufficiently accurate for your needs. It may need additional training to further improve it.
-newckpt = cpdir + '/checkpoint.pt'
+newckpt = cpdir + '/checkpoint.pt'
newcalc = OCPCalculator(checkpoint_path=newckpt, cpu=False)
-
-WARNING:root:Unrecognized arguments: ['symmetric_edge_symmetrization']
-
-
-WARNING:root:Using `weight_decay` from `optim` instead of `optim.optimizer_params`.Please update your config to use `optim.optimizer_params.weight_decay`.`optim.weight_decay` will soon be deprecated.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-
-eos_data = {}
+eos_data = {}
for oxide in oxides:
eos_data[oxide] = {}
for polymorph in polymorphs:
@@ -1263,27 +953,19 @@ Running the training job
-
-
-mae = np.mean(np.abs(np.array(dft) - np.array(ocp)))
+mae = np.mean(np.abs(np.array(dft) - np.array(ocp)))
print(f'New MAE = {mae:1.3f} eV/atom')
-
-New MAE = 0.549 eV/atom
-
-
-
This is a substantial improvement! We can see what it means in terms of the Equations of State. There is still not perfect agreement, but the curves are closer together. Additional fine tuning, or a better model could probably still improve this.
-oxide, polymorph = 'VO2', 'fluorite'
+oxide, polymorph = 'VO2', 'fluorite'
V, D, O = eos_data[oxide][polymorph]
plt.plot(V, D, label='dft')
@@ -1293,9 +975,6 @@ Running the training job
-
-
It is possible to continue refining the fit. The simple things to do are to use more epochs of training. Eventually the MAE will stabilize, and then it may be necessary to adjust other optimization parameters like the learning rate (usually you decrease it).
Depending on what is important to you, you may consider changing the relative importances of energy and forces; you can often trade off accuracy of one for the other. It may be necessary to add additional data to cover the composition and configuration space more thoroughly.
diff --git a/core/gotchas.html b/core/gotchas.html
index 9c6f941b30..a1e6906b28 100644
--- a/core/gotchas.html
+++ b/core/gotchas.html
@@ -8,7 +8,7 @@
- Common gotchas with fairchem — FAIR Chemistry Documentation
+ Common gotchas with fairchem
@@ -146,14 +146,8 @@
-
-
-
-
-
-
-
+
@@ -173,7 +167,7 @@
@@ -666,31 +660,17 @@ I want the energy of a gas phase atom
-from fairchem.core.common.relaxation.ase_utils import OCPCalculator
+from fairchem.core.common.relaxation.ase_utils import OCPCalculator
from fairchem.core.models.model_registry import model_name_to_local_file
checkpoint_path = model_name_to_local_file('GemNet-OC-S2EFS-OC20+OC22', local_cache='/tmp/ocp_checkpoints/')
calc = OCPCalculator(checkpoint_path=checkpoint_path)
-
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Unrecognized arguments: ['symmetric_edge_symmetrization']
-
-
-WARNING:root:Using `weight_decay` from `optim` instead of `optim.optimizer_params`.Please update your config to use `optim.optimizer_params.weight_decay`.`optim.weight_decay` will soon be deprecated.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-
-%%capture
+
@@ -875,7 +686,7 @@ I get wildly different energies from the different models
-# These are to suppress the output from making the calculators.
+# These are to suppress the output from making the calculators.
from io import StringIO
import contextlib
@@ -884,7 +695,7 @@ I get wildly different energies from the different models
-from ase.build import fcc111, add_adsorbate
+from ase.build import fcc111, add_adsorbate
slab = fcc111('Pt', size=(2, 2, 5), vacuum=10.0)
add_adsorbate(slab, 'O', height=1.2, position='fcc')
@@ -893,7 +704,7 @@ I get wildly different energies from the different models
-from fairchem.core.models.model_registry import model_name_to_local_file
+from fairchem.core.models.model_registry import model_name_to_local_file
# OC20 model - trained on adsorption energies
checkpoint_path = model_name_to_local_file('GemNet-OC-S2EF-OC20-All', local_cache='/tmp/ocp_checkpoints/')
@@ -908,27 +719,10 @@ I get wildly different energies from the different models
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Unrecognized arguments: ['symmetric_edge_symmetrization']
-
-
-WARNING:root:Using `weight_decay` from `optim` instead of `optim.optimizer_params`.Please update your config to use `optim.optimizer_params.weight_decay`.`optim.weight_decay` will soon be deprecated.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-1.2779738903045654
-
-
-
-# An OC22 checkpoint - trained on total energy
+# An OC22 checkpoint - trained on total energy
checkpoint_path = model_name_to_local_file('GemNet-OC-S2EFS-OC20+OC22', local_cache='/tmp/ocp_checkpoints/')
with contextlib.redirect_stdout(StringIO()) as _:
@@ -941,27 +735,10 @@ I get wildly different energies from the different models
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Unrecognized arguments: ['symmetric_edge_symmetrization']
-
-
-WARNING:root:Using `weight_decay` from `optim` instead of `optim.optimizer_params`.Please update your config to use `optim.optimizer_params.weight_decay`.`optim.weight_decay` will soon be deprecated.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
--110.3986587524414
-
-
-
-# This eSCN model is trained on adsorption energies
+# This eSCN model is trained on adsorption energies
checkpoint_path = model_name_to_local_file('eSCN-L4-M2-Lay12-S2EF-OC20-2M', local_cache='/tmp/ocp_checkpoints/')
with contextlib.redirect_stdout(StringIO()) as _:
@@ -972,17 +749,6 @@ I get wildly different energies from the different models
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-1.6730773448944092
-
-
-
@@ -1015,7 +781,7 @@ You need at least four atoms for molecules with some models
-%%capture
+%%capture
from fairchem.core.common.relaxation.ase_utils import OCPCalculator
from fairchem.core.models.model_registry import model_name_to_local_file
import os
@@ -1026,24 +792,10 @@ You need at least four atoms for molecules with some models
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Unrecognized arguments: ['symmetric_edge_symmetrization']
-
-
-WARNING:root:Using `weight_decay` from `optim` instead of `optim.optimizer_params`.Please update your config to use `optim.optimizer_params.weight_decay`.`optim.weight_decay` will soon be deprecated.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-
-%%capture
+
@@ -1230,7 +813,7 @@ To tag or not?
-%%capture
+%%capture
from fairchem.core.common.relaxation.ase_utils import OCPCalculator
from fairchem.core.models.model_registry import model_name_to_local_file
import os
@@ -1240,219 +823,31 @@ To tag or not?
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Unrecognized arguments: ['symmetric_edge_symmetrization']
-
-
-WARNING:root:Using `weight_decay` from `optim` instead of `optim.optimizer_params`.Please update your config to use `optim.optimizer_params.weight_decay`.`optim.weight_decay` will soon be deprecated.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-
-%%capture
+%%capture
atoms = molecule('CH4')
atoms.set_calculator(calc)
atoms.get_potential_energy() # error
-
----------------------------------------------------------------------------
-RuntimeError Traceback (most recent call last)
-Cell In[11], line 3
- 1 atoms = molecule('CH4')
- 2 atoms.set_calculator(calc)
-----> 3 atoms.get_potential_energy() # error
-
-File /opt/hostedtoolcache/Python/3.11.9/x64/lib/python3.11/site-packages/ase/atoms.py:731, in Atoms.get_potential_energy(self, force_consistent, apply_constraint)
- 728 energy = self._calc.get_potential_energy(
- 729 self, force_consistent=force_consistent)
- 730 else:
---> 731 energy = self._calc.get_potential_energy(self)
- 732 if apply_constraint:
- 733 for constraint in self.constraints:
-
-File /opt/hostedtoolcache/Python/3.11.9/x64/lib/python3.11/site-packages/ase/calculators/calculator.py:709, in Calculator.get_potential_energy(self, atoms, force_consistent)
- 708 def get_potential_energy(self, atoms=None, force_consistent=False):
---> 709 energy = self.get_property('energy', atoms)
- 710 if force_consistent:
- 711 if 'free_energy' not in self.results:
-
-File /opt/hostedtoolcache/Python/3.11.9/x64/lib/python3.11/site-packages/ase/calculators/calculator.py:737, in Calculator.get_property(self, name, atoms, allow_calculation)
- 735 if not allow_calculation:
- 736 return None
---> 737 self.calculate(atoms, [name], system_changes)
- 739 if name not in self.results:
- 740 # For some reason the calculator was not able to do what we want,
- 741 # and that is OK.
- 742 raise PropertyNotImplementedError('{} not present in this '
- 743 'calculation'.format(name))
-
-File ~/work/fairchem/fairchem/src/fairchem/core/common/relaxation/ase_utils.py:243, in OCPCalculator.calculate(self, atoms, properties, system_changes)
- 240 data_object = self.a2g.convert(atoms)
- 241 batch = data_list_collater([data_object], otf_graph=True)
---> 243 predictions = self.trainer.predict(batch, per_image=False, disable_tqdm=True)
- 245 for key in predictions:
- 246 _pred = predictions[key]
-
-File /opt/hostedtoolcache/Python/3.11.9/x64/lib/python3.11/site-packages/torch/utils/_contextlib.py:115, in context_decorator.<locals>.decorate_context(*args, **kwargs)
- 112 @functools.wraps(func)
- 113 def decorate_context(*args, **kwargs):
- 114 with ctx_factory():
---> 115 return func(*args, **kwargs)
-
-File ~/work/fairchem/fairchem/src/fairchem/core/trainers/ocp_trainer.py:434, in OCPTrainer.predict(self, data_loader, per_image, results_file, disable_tqdm)
- 426 for _i, batch in tqdm(
- 427 enumerate(data_loader),
- 428 total=len(data_loader),
- (...)
- 431 disable=disable_tqdm,
- 432 ):
- 433 with torch.cuda.amp.autocast(enabled=self.scaler is not None):
---> 434 out = self._forward(batch)
- 436 for target_key in self.config["outputs"]:
- 437 pred = out[target_key]
-
-File ~/work/fairchem/fairchem/src/fairchem/core/trainers/ocp_trainer.py:235, in OCPTrainer._forward(self, batch)
- 234 def _forward(self, batch):
---> 235 out = self.model(batch.to(self.device))
- 237 ### TODO: Move into BaseModel in OCP 2.0
- 238 outputs = {}
-
-File /opt/hostedtoolcache/Python/3.11.9/x64/lib/python3.11/site-packages/torch/nn/modules/module.py:1511, in Module._wrapped_call_impl(self, *args, **kwargs)
- 1509 return self._compiled_call_impl(*args, **kwargs) # type: ignore[misc]
- 1510 else:
--> 1511 return self._call_impl(*args, **kwargs)
-
-File /opt/hostedtoolcache/Python/3.11.9/x64/lib/python3.11/site-packages/torch/nn/modules/module.py:1520, in Module._call_impl(self, *args, **kwargs)
- 1515 # If we don't have any hooks, we want to skip the rest of the logic in
- 1516 # this function, and just call forward.
- 1517 if not (self._backward_hooks or self._backward_pre_hooks or self._forward_hooks or self._forward_pre_hooks
- 1518 or _global_backward_pre_hooks or _global_backward_hooks
- 1519 or _global_forward_hooks or _global_forward_pre_hooks):
--> 1520 return forward_call(*args, **kwargs)
- 1522 try:
- 1523 result = None
-
-File ~/work/fairchem/fairchem/src/fairchem/core/common/utils.py:156, in conditional_grad.<locals>.decorator.<locals>.cls_method(self, *args, **kwargs)
- 154 if self.regress_forces and not getattr(self, "direct_forces", 0):
- 155 f = dec(func)
---> 156 return f(self, *args, **kwargs)
-
-File ~/work/fairchem/fairchem/src/fairchem/core/models/gemnet_oc/gemnet_oc.py:1229, in GemNetOC.forward(self, data)
- 1207 (
- 1208 main_graph,
- 1209 a2a_graph,
- (...)
- 1216 quad_idx,
- 1217 ) = self.get_graphs_and_indices(data)
- 1218 _, idx_t = main_graph["edge_index"]
- 1220 (
- 1221 basis_rad_raw,
- 1222 basis_atom_update,
- 1223 basis_output,
- 1224 bases_qint,
- 1225 bases_e2e,
- 1226 bases_a2e,
- 1227 bases_e2a,
- 1228 basis_a2a_rad,
--> 1229 ) = self.get_bases(
- 1230 main_graph=main_graph,
- 1231 a2a_graph=a2a_graph,
- 1232 a2ee2a_graph=a2ee2a_graph,
- 1233 qint_graph=qint_graph,
- 1234 trip_idx_e2e=trip_idx_e2e,
- 1235 trip_idx_a2e=trip_idx_a2e,
- 1236 trip_idx_e2a=trip_idx_e2a,
- 1237 quad_idx=quad_idx,
- 1238 num_atoms=num_atoms,
- 1239 )
- 1241 # Embedding block
- 1242 h = self.atom_emb(atomic_numbers)
-
-File ~/work/fairchem/fairchem/src/fairchem/core/models/gemnet_oc/gemnet_oc.py:1102, in GemNetOC.get_bases(self, main_graph, a2a_graph, a2ee2a_graph, qint_graph, trip_idx_e2e, trip_idx_a2e, trip_idx_e2a, quad_idx, num_atoms)
- 1093 cosφ_cab_q, cosφ_abd, angle_cabd = self.calculate_quad_angles(
- 1094 main_graph["vector"],
- 1095 qint_graph["vector"],
- 1096 quad_idx,
- 1097 )
- 1099 basis_rad_cir_qint_raw, basis_cir_qint_raw = self.cbf_basis_qint(
- 1100 qint_graph["distance"], cosφ_abd
- 1101 )
--> 1102 basis_rad_sph_qint_raw, basis_sph_qint_raw = self.sbf_basis_qint(
- 1103 main_graph["distance"],
- 1104 cosφ_cab_q[quad_idx["trip_out_to_quad"]],
- 1105 angle_cabd,
- 1106 )
- 1107 if self.atom_edge_interaction:
- 1108 basis_rad_a2ee2a_raw = self.radial_basis_aeaint(a2ee2a_graph["distance"])
-
-File /opt/hostedtoolcache/Python/3.11.9/x64/lib/python3.11/site-packages/torch/nn/modules/module.py:1511, in Module._wrapped_call_impl(self, *args, **kwargs)
- 1509 return self._compiled_call_impl(*args, **kwargs) # type: ignore[misc]
- 1510 else:
--> 1511 return self._call_impl(*args, **kwargs)
-
-File /opt/hostedtoolcache/Python/3.11.9/x64/lib/python3.11/site-packages/torch/nn/modules/module.py:1520, in Module._call_impl(self, *args, **kwargs)
- 1515 # If we don't have any hooks, we want to skip the rest of the logic in
- 1516 # this function, and just call forward.
- 1517 if not (self._backward_hooks or self._backward_pre_hooks or self._forward_hooks or self._forward_pre_hooks
- 1518 or _global_backward_pre_hooks or _global_backward_hooks
- 1519 or _global_forward_hooks or _global_forward_pre_hooks):
--> 1520 return forward_call(*args, **kwargs)
- 1522 try:
- 1523 result = None
-
-File ~/work/fairchem/fairchem/src/fairchem/core/models/gemnet_oc/layers/spherical_basis.py:132, in SphericalBasisLayer.forward(self, D_ca, cosφ_cab, θ_cabd)
- 130 def forward(self, D_ca, cosφ_cab, θ_cabd):
- 131 rad_basis = self.radial_basis(D_ca)
---> 132 sph_basis = self.spherical_basis(cosφ_cab, θ_cabd)
- 133 # (num_quadruplets, num_spherical**2)
- 135 if self.scale_basis:
-
-File ~/work/fairchem/fairchem/src/fairchem/core/models/gemnet_oc/layers/spherical_basis.py:116, in SphericalBasisLayer.__init__.<locals>.<lambda>(cosφ, θ)
- 111 elif sbf_name == "legendre_outer":
- 112 circular_basis = get_sph_harm_basis(num_spherical, zero_m_only=True)
- 113 self.spherical_basis = lambda cosφ, ϑ: (
- 114 circular_basis(cosφ)[:, :, None]
- 115 * circular_basis(torch.cos(ϑ))[:, None, :]
---> 116 ).reshape(cosφ.shape[0], -1)
- 118 elif sbf_name == "gaussian_outer":
- 119 self.circular_basis = GaussianBasis(
- 120 start=-1, stop=1, num_gaussians=num_spherical, **sbf_hparams
- 121 )
-
-RuntimeError: cannot reshape tensor of 0 elements into shape [0, -1] because the unspecified dimension size -1 can be any value and is ambiguous
-
-
-
-atoms = molecule('CH4')
+atoms = molecule('CH4')
atoms.set_tags(np.ones(len(atoms))) # <- critical line for Gemnet
atoms.set_calculator(calc)
atoms.get_potential_energy()
-
--23.71796226501465
-
-
-
Not all models require tags though. This EquiformerV2 model does not use them. This is another detail that is important to keep in mind.
-from fairchem.core.common.relaxation.ase_utils import OCPCalculator
+from fairchem.core.common.relaxation.ase_utils import OCPCalculator
from fairchem.core.models.model_registry import model_name_to_local_file
import os
@@ -1462,32 +857,16 @@ To tag or not?
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Skipping scheduler setup. No training set found.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-
-atoms = molecule('CH4')
+atoms = molecule('CH4')
atoms.set_calculator(calc)
atoms.get_potential_energy()
-
--0.42973753809928894
-
-
-
@@ -1497,7 +876,7 @@ Stochastic simulation results
-from fairchem.core.models.model_registry import model_name_to_local_file
+from fairchem.core.models.model_registry import model_name_to_local_file
from fairchem.core.common.relaxation.ase_utils import OCPCalculator
checkpoint_path = model_name_to_local_file('EquiformerV2-31M-S2EF-OC20-All+MD', local_cache='/tmp/ocp_checkpoints/')
@@ -1521,30 +900,6 @@ Stochastic simulation results
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Skipping scheduler setup. No training set found.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-1.2139877557754517 7.235522701975592e-07
-1.2139873504638672
-1.2139885425567627
-1.2139878273010254
-1.2139883041381836
-1.2139863967895508
-1.2139866352081299
-1.2139875888824463
-1.2139883041381836
-1.2139880657196045
-1.2139885425567627
-
-
-
@@ -1552,7 +907,7 @@ The forces don’t sum to zero
-from fairchem.core.models.model_registry import model_name_to_local_file
+from fairchem.core.models.model_registry import model_name_to_local_file
checkpoint_path = model_name_to_local_file('EquiformerV2-31M-S2EF-OC20-All+MD', local_cache='/tmp/ocp_checkpoints/')
from fairchem.core.common.relaxation.ase_utils import OCPCalculator
@@ -1569,33 +924,14 @@ The forces don’t sum to zero
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Skipping scheduler setup. No training set found.
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-array([ 0.00847984, 0.01409346, -0.05882812], dtype=float32)
-
-
-
-# This makes them sum closer to zero by removing net translational force
+# This makes them sum closer to zero by removing net translational force
(f - f.mean(axis=0)).sum(axis=0)
-
-array([1.2258533e-07, 4.1909516e-09, 1.1920929e-07], dtype=float32)
-
-
-
diff --git a/core/inference.html b/core/inference.html
index 702aae2dfd..7f2cbcde48 100644
--- a/core/inference.html
+++ b/core/inference.html
@@ -8,7 +8,7 @@
- Fast batched inference — FAIR Chemistry Documentation
+ Fast batched inference
@@ -146,14 +146,8 @@
-
-
-
-
-
-
-
+
@@ -173,7 +167,7 @@
@@ -640,67 +634,7 @@ Fast batched inferenceYou can retrieve the dataset below. In this notebook we learn how to do “mass inference” without an ASE calculator. You do this by creating a config.yml file, and running the main.py command line utility.
-! wget https://figshare.com/ndownloader/files/11948267 -O data.db
-
-
-
-
---2024-05-17 16:46:43-- https://figshare.com/ndownloader/files/11948267
-Resolving figshare.com (figshare.com)...
-
-
-52.214.121.144, 63.33.12.55, 2a05:d018:1f4:d000:2dfa:52ce:c4c9:d519, ...
-Connecting to figshare.com (figshare.com)|52.214.121.144|:443...
-
-
-connected.
-
-
-HTTP request sent, awaiting response...
-
-
-302 Found
-Location: https://s3-eu-west-1.amazonaws.com/pstorage-cmu-348901238291901/11948267/data.db?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAI266R7V6O36O5JUA/20240517/eu-west-1/s3/aws4_request&X-Amz-Date=20240517T164644Z&X-Amz-Expires=10&X-Amz-SignedHeaders=host&X-Amz-Signature=2c9b151c97ccdb699d74fbcdcb3fbe6c349d7919813063b31b67939734cfad93 [following]
---2024-05-17 16:46:44-- https://s3-eu-west-1.amazonaws.com/pstorage-cmu-348901238291901/11948267/data.db?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAI266R7V6O36O5JUA/20240517/eu-west-1/s3/aws4_request&X-Amz-Date=20240517T164644Z&X-Amz-Expires=10&X-Amz-SignedHeaders=host&X-Amz-Signature=2c9b151c97ccdb699d74fbcdcb3fbe6c349d7919813063b31b67939734cfad93
-Resolving s3-eu-west-1.amazonaws.com (s3-eu-west-1.amazonaws.com)... 52.218.117.176, 52.218.116.8, 52.218.45.0, ...
-Connecting to s3-eu-west-1.amazonaws.com (s3-eu-west-1.amazonaws.com)|52.218.117.176|:443...
-
-
-connected.
-
-
-HTTP request sent, awaiting response...
-
-
-200 OK
-Length: 43125760 (41M) [binary/octet-stream]
-Saving to: ‘data.db’
-
-
-data.db 0%[ ] 0 --.-KB/s
-
-
-data.db 0%[ ] 142.38K 660KB/s
-
-
-data.db 3%[ ] 1.27M 2.94MB/s
-
-
-data.db 19%[==> ] 8.04M 12.4MB/s
-
-
-data.db 40%[=======> ] 16.73M 19.4MB/s
-
-
-data.db 61%[===========> ] 25.19M 23.3MB/s
-
-
-data.db 82%[===============> ] 33.73M 26.2MB/s
-
-
-data.db 100%[===================>] 41.13M 28.1MB/s in 1.5s
-
-2024-05-17 16:46:46 (28.1 MB/s) - ‘data.db’ saved [43125760/43125760]
+! wget https://figshare.com/ndownloader/files/11948267 -O data.db
@@ -709,7 +643,7 @@ Fast batched inference
-! mv data.db full_data.db
+! mv data.db full_data.db
import ase.db
import numpy as np
@@ -731,34 +665,7 @@ Fast batched inference
-! ase db data.db
-
-
-
-
-id|age|user |formula|calculator| energy|natoms| fmax|pbc| volume|charge| mass
- 1| 0s|runner|Au55 |unknown |-136.105| 55|1.019|TTT|6865.782| 0.000|10833.161
- 2| 0s|runner|Au30 |unknown | -68.473| 30|0.935|TTT|5253.718| 0.000| 5908.997
- 3| 0s|runner|Au107 |unknown |-344.348| 107|0.005|TTT|1921.504| 0.000|21075.423
- 4| 0s|runner|Au2 |unknown | -5.225| 2|0.000|TTT| 50.970| 0.000| 393.933
- 5| 0s|runner|Au18 |unknown | -43.660| 18|0.290|TTT|4006.041| 0.000| 3545.398
- 6| 0s|runner|Au4 |unknown | -10.130| 4|9.234|TTT| 156.940| 0.000| 787.866
- 7| 0s|runner|Au4 |unknown | -12.085| 4|0.279|TTT| 143.096| 0.000| 787.866
- 8| 0s|runner|Au4 |unknown | -12.198| 4|0.047|TTT| 143.096| 0.000| 787.866
- 9| 0s|runner|Au26 |unknown | -64.273| 26|0.267|TTT|4932.626| 0.000| 5121.131
-10| 0s|runner|Au52 |unknown |-135.782| 52|0.360|TTT|6744.510| 0.000|10242.262
-11| 0s|runner|Au26 |unknown | -82.667| 26|0.375|TTT| 473.514| 0.000| 5121.131
-12| 0s|runner|Au30 |unknown | -75.754| 30|0.106|TTT|5253.718| 0.000| 5908.997
-13| 0s|runner|Au38 |unknown |-100.379| 38|0.310|TTT|5846.048| 0.000| 7484.730
-14| 0s|runner|Au54 |unknown |-140.559| 54|0.347|TTT|6906.090| 0.000|10636.195
-15| 0s|runner|Au12 |unknown | -26.924| 12|0.194|TTT|3571.095| 0.000| 2363.599
-16| 0s|runner|Au55 |unknown |-148.793| 55|0.369|TTT|7781.809| 0.000|10833.161
-17| 0s|runner|Au12 |unknown | -26.510| 12|0.645|TTT|3571.095| 0.000| 2363.599
-18| 0s|runner|Au4 |unknown | -11.727| 4|0.170|TTT| 156.940| 0.000| 787.866
-19| 0s|runner|Au43 |unknown |-104.589| 43|0.444|TTT|6467.812| 0.000| 8469.562
-20| 0s|runner|Au17 |unknown | -40.840| 17|0.231|TTT|4006.041| 0.000| 3348.432
-Rows: 50 (showing first 20)
-Keys: NEB, bulk, calc_time, cluster, concentration, config, converged, diffusion, ediff, ... (29 more)
+! ase db data.db
@@ -766,36 +673,26 @@ Fast batched inferenceYou have to choose a checkpoint to start with. The newer checkpoints may require too much memory for this environment.
-from fairchem.core.models.model_registry import available_pretrained_models
+from fairchem.core.models.model_registry import available_pretrained_models
print(available_pretrained_models)
-
-('CGCNN-S2EF-OC20-200k', 'CGCNN-S2EF-OC20-2M', 'CGCNN-S2EF-OC20-20M', 'CGCNN-S2EF-OC20-All', 'DimeNet-S2EF-OC20-200k', 'DimeNet-S2EF-OC20-2M', 'SchNet-S2EF-OC20-200k', 'SchNet-S2EF-OC20-2M', 'SchNet-S2EF-OC20-20M', 'SchNet-S2EF-OC20-All', 'DimeNet++-S2EF-OC20-200k', 'DimeNet++-S2EF-OC20-2M', 'DimeNet++-S2EF-OC20-20M', 'DimeNet++-S2EF-OC20-All', 'SpinConv-S2EF-OC20-2M', 'SpinConv-S2EF-OC20-All', 'GemNet-dT-S2EF-OC20-2M', 'GemNet-dT-S2EF-OC20-All', 'PaiNN-S2EF-OC20-All', 'GemNet-OC-S2EF-OC20-2M', 'GemNet-OC-S2EF-OC20-All', 'GemNet-OC-S2EF-OC20-All+MD', 'GemNet-OC-Large-S2EF-OC20-All+MD', 'SCN-S2EF-OC20-2M', 'SCN-t4-b2-S2EF-OC20-2M', 'SCN-S2EF-OC20-All+MD', 'eSCN-L4-M2-Lay12-S2EF-OC20-2M', 'eSCN-L6-M2-Lay12-S2EF-OC20-2M', 'eSCN-L6-M2-Lay12-S2EF-OC20-All+MD', 'eSCN-L6-M3-Lay20-S2EF-OC20-All+MD', 'EquiformerV2-83M-S2EF-OC20-2M', 'EquiformerV2-31M-S2EF-OC20-All+MD', 'EquiformerV2-153M-S2EF-OC20-All+MD', 'SchNet-S2EF-force-only-OC20-All', 'DimeNet++-force-only-OC20-All', 'DimeNet++-Large-S2EF-force-only-OC20-All', 'DimeNet++-S2EF-force-only-OC20-20M+Rattled', 'DimeNet++-S2EF-force-only-OC20-20M+MD', 'CGCNN-IS2RE-OC20-10k', 'CGCNN-IS2RE-OC20-100k', 'CGCNN-IS2RE-OC20-All', 'DimeNet-IS2RE-OC20-10k', 'DimeNet-IS2RE-OC20-100k', 'DimeNet-IS2RE-OC20-all', 'SchNet-IS2RE-OC20-10k', 'SchNet-IS2RE-OC20-100k', 'SchNet-IS2RE-OC20-All', 'DimeNet++-IS2RE-OC20-10k', 'DimeNet++-IS2RE-OC20-100k', 'DimeNet++-IS2RE-OC20-All', 'PaiNN-IS2RE-OC20-All', 'GemNet-dT-S2EFS-OC22', 'GemNet-OC-S2EFS-OC22', 'GemNet-OC-S2EFS-OC20+OC22', 'GemNet-OC-S2EFS-nsn-OC20+OC22', 'GemNet-OC-S2EFS-OC20->OC22', 'EquiformerV2-lE4-lF100-S2EFS-OC22', 'SchNet-S2EF-ODAC', 'DimeNet++-S2EF-ODAC', 'PaiNN-S2EF-ODAC', 'GemNet-OC-S2EF-ODAC', 'eSCN-S2EF-ODAC', 'EquiformerV2-S2EF-ODAC', 'EquiformerV2-Large-S2EF-ODAC', 'Gemnet-OC-IS2RE-ODAC', 'eSCN-IS2RE-ODAC', 'EquiformerV2-IS2RE-ODAC')
-
-
-
-from fairchem.core.models.model_registry import model_name_to_local_file
+from fairchem.core.models.model_registry import model_name_to_local_file
checkpoint_path = model_name_to_local_file('GemNet-dT-S2EFS-OC22', local_cache='/tmp/ocp_checkpoints/')
checkpoint_path
-
-'/tmp/ocp_checkpoints/gndt_oc22_all_s2ef.pt'
-
-
-
We have to update our configuration yml file with the dataset. It is necessary to specify the train and test set for some reason.
-from fairchem.core.common.tutorial_utils import generate_yml_config
+
-
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Scale factor comment not found in model
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-PosixPath('/home/runner/work/fairchem/fairchem/docs/core/config.yml')
-
-
-
It is a good idea to redirect the output to a file. If the output gets too large here, the notebook may fail to save. Normally I would use a redirect like 2&>1, but this does not work with the main.py method. An alternative here is to open a terminal and run it there.
-%%capture inference
+%%capture inference
import time
from fairchem.core.common.tutorial_utils import fairchem_main
@@ -851,7 +734,7 @@ Fast batched inference
-with open('mass-inference.txt', 'wb') as f:
+with open('mass-inference.txt', 'wb') as f:
f.write(inference.stdout.encode('utf-8'))
@@ -859,20 +742,15 @@ Fast batched inference
-! grep "Total time taken:" 'mass-inference.txt'
-
-
-
-
-2024-05-17 16:47:05 (INFO): Total time taken: 7.855764865875244
+! grep "Total time taken:" 'mass-inference.txt'
-The mass inference approach takes 1-2 minutes to run. See the output here.
+The mass inference approach takes 1-2 minutes to run. See the output here.
-results = ! grep " results_dir:" mass-inference.txt
+results = ! grep " results_dir:" mass-inference.txt
d = results[0].split(':')[-1].strip()
@@ -880,39 +758,27 @@ Fast batched inference
-import numpy as np
+import numpy as np
results = np.load(f'{d}/ocp_predictions.npz', allow_pickle=True)
results.files
-
-['energy', 'forces', 'chunk_idx', 'ids']
-
-
-
It is not obvious, but the data from mass inference is not in the same order. We have to get an id from the mass inference, and then “resort” the results so they are in the same order.
-inds = np.array([int(r.split('_')[0]) for r in results['ids']])
+inds = np.array([int(r.split('_')[0]) for r in results['ids']])
sind = np.argsort(inds)
inds[sind]
-
-array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
- 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33,
- 34, 35, 36, 37, 38])
-
-
-
To compare this with the results, we need to get the energy data from the ase db.
-from ase.db import connect
+from ase.db import connect
db = connect('data.db')
energies = np.array([row.energy for row in db.select('natoms>5,xc=PBE')])
@@ -924,7 +790,7 @@ Fast batched inferenceNow, we can see the predictions. They are only ok here; that is not surprising, the data set has lots of Au configurations that have never been seen by this model. Fine-tuning would certainly help improve this.
-import matplotlib.pyplot as plt
+
-
-
-
@@ -946,26 +809,15 @@ The ASE calculator way
-from fairchem.core.common.relaxation.ase_utils import OCPCalculator
+from fairchem.core.common.relaxation.ase_utils import OCPCalculator
calc = OCPCalculator(checkpoint_path=checkpoint_path, cpu=False)
-
-WARNING:root:Detected old config, converting to new format. Consider updating to avoid potential incompatibilities.
-
-
-WARNING:root:Scale factor comment not found in model
-
-
-WARNING:root:No seed has been set in modelcheckpoint or OCPCalculator! Results may not be reproducible on re-run
-
-
-
-import time
+