4.6. Descriptor DPA-2 PyTorch DP

Note

Supported backends: PyTorch PyTorch, DP DP

The DPA-2 model implementation. See https://arxiv.org/abs/2312.15492 for more details.

Training example: examples/water/dpa2/input_torch.json.

4.6.1. Data format

DPA-2 supports both the standard data format and the mixed type data format.