FINETUNA accelerates atomistic simulations by fine-tuning a pre-trained graph model in an active learning framework.
Installation is easy:
conda env create -f env.cpu.yml
conda activate finetuna
cd finetuna
pip install -e .
git clone https://github.com/Open-Catalyst-Project/ocp.git
cd ocp
pip install -e .
pip install git+https://github.com/ulissigroup/vasp-interactive.git
All pre-trained machine learning model checkpoint can be found here. We recommend to download the GemNet-dT all model. click here to download.
You are all set! Now in your VASP input folder, run the calculation by: finetuna_wrap.py -c /path/to/the/checkpoint
.
If you have an ASE atoms object, see example 1 and 2.
If you have VASP input files (INCAR, KPOINTS, POTCAR, and POSCAR), see example 3.