DPA-3.1-3M is trained using a multitask strategy on the OpenLAM datasets. The benchmark results can be found at the LAMBench website.

Please ensure the usage of the correct code version (v3.1.0) corresponding to this model.

To use the pretrained model, you need to first install the corresponding version of DeePMD-kit. You can try easy installation:

For other installation options, please visit the Releases page to download the off-line package for deepmd-kit devel, and refer to the official documentation for off-line installation instructions.

The pretrained model can be used directly for prediction tasks, such as serving as an ASE calculator.

This model is pretrained in a multi-task manner, featuring a unified backbone (referred to as the unified descriptor and the unified fitting net) and an dataset encoding vector for different tasks. For detailed information, refer to the DPA-3 paper.

The first step involves selecting a specific dataset id from the model to make predictions. To list the available fitting nets (model-branch) in this pretrained model, use the following command:

This will generate an output similar to the following:

Select a dataset id that closely matches your system. Ensure that the elements in your system are included in the corresponding task if you are conducting direct predictions or simulations. For more information on the pretrained datasets, refer to below.

Assuming you choose H2O_H2O-PD, you can first freeze the model branch from the multi-task pretrained model:

Then you can use the following Python code for prediction or optimization:

For more advanced usages like fine-tuning and zero-shot inference, please refer to the following model card .