search

deepmodeling / deepmd-kit

A deep learning package for many-body potential energy representation and molecular dynamics
https://docs.deepmodeling.com/projects/deepmd/
GNU Lesser General Public License v3.0
1.41k stars 487 forks source link

How to restart DPA-2 training? #3819

Closed zhbzhbzhbzhb closed 1 month ago

zhbzhbzhbzhb commented 1 month ago

Summary

How to restart PDA-2 training?

DeePMD-kit Version

3.0.0a1.dev82+geed7c8ac

Backend and its version

torch2.0.0 torchvision 0.15.1 torchaudio-2.0.1

Python Version, CUDA Version, GCC Version, LAMMPS Version, etc

No response

Details

Should I use: dp train --restart model.ckpt input.json --finetune OpenLAM_2.1.0_27heads_2024Q1.pt --model-branch Domains_OC2M