Skip to content

add DPA3 7M model #101

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
anyangml merged 2 commits into from
Mar 3, 2025
Merged

add DPA3 7M model #101

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
66a8fb3
add DPA3 7M model
caic99 Mar 3, 2025
039a0c6
Update deepmd-kit source code branch
caic99 Mar 3, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions lambench/models/models_config.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,17 @@
show_finetune_task: True
show_calculator_task: True

- model_name: dpa3_250211_v3_0_0_7M
model_type: DP
model_family: DP
model_path: /bohr/lambench-model-55c1/v4/dpa3_250211_v3_0_0_7M/dp_dpa3_v3.0.0_0211_280w.pt
virtualenv: registry.dp.tech/dptech/dp/native/prod-26832/lambench:v2.2
model_metadata:
model_description: "DP 2025 Q1, 7M parameters"
show_finetune_task: False
show_calculator_task: False
skip_tasks: [PropertyFinetuneTask, CalculatorTask]

- model_name: mace_mp_0_medium
model_type: ASE
model_family: MACE
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ classifiers = [
]

[project.optional-dependencies]
deepmd = ["deepmd-kit @ git+https://github.com/iProzd/deepmd-kit.git@0103_prob_scp_e1a#egg=deepmd-kit", "torch>=2.1"]
deepmd = ["deepmd-kit @ git+https://github.com/iProzd/deepmd-kit.git@devel_dpa3_descriptor#egg=deepmd-kit", "torch>=2.1"]
fairchem = ["fairchem-core", "torch_geometric", "pyg_lib", "torch_scatter", "torch_sparse", "torch_cluster", "torch_spline_conv"] # See README
mace = ["mace-torch"]
mattersim = ["mattersim @ git+https://github.com/microsoft/mattersim.git@f1028e5eee8cc19edfc1d44f3c468c1206192daa"]
Expand Down