Skip to content

Commit cbae3c9

Browse files
committed
feat: add dpa3_paper_model_tests
1 parent b1bbede commit cbae3c9

File tree

1 file changed

+41
-0
lines changed

1 file changed

+41
-0
lines changed

lambench/models/models_config.yml

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,3 +52,44 @@
5252
num_parameters: 6637036
5353
packages:
5454
deepmd-kit: P0131
55+
56+
- model_name: dpa3_250415_best_XC
57+
model_type: DP
58+
model_family: DP
59+
model_path: /bohr/testmodel-gevy/v2/test_0415/model.ckpt-3200000.pt
60+
virtualenv: registry.dp.tech/dptech/dp/native/prod-375/lambench:v2.11
61+
model_metadata:
62+
pretty_name: dpa3_250415_best_XC
63+
date_added: 2025-04-01
64+
model_description: DP 2024 Q4, 37head share fitting 120GPU pretrain
65+
num_parameters: 6637036
66+
packages:
67+
deepmd-kit: P0131
68+
69+
70+
- model_name: dpa3_250420_50w
71+
model_type: DP
72+
model_family: DP
73+
model_path: /bohr/testmodel-gevy/v3/test_dpa3/model.ckpt-500000.pt
74+
virtualenv: registry.dp.tech/dptech/dp/native/prod-375/lambench:v2.11
75+
model_metadata:
76+
pretty_name: dpa3_250420_50w
77+
date_added: 2025-04-01
78+
model_description: DP 2024 Q4, 37head share fitting 120GPU pretrain
79+
num_parameters: 6637036
80+
packages:
81+
deepmd-kit: P0131
82+
83+
84+
- model_name: dpa3_250420_200w
85+
model_type: DP
86+
model_family: DP
87+
model_path: /bohr/testmodel-gevy/v3/test_dpa3/model.ckpt-2000000.pt
88+
virtualenv: registry.dp.tech/dptech/dp/native/prod-375/lambench:v2.11
89+
model_metadata:
90+
pretty_name: dpa3_250420_200w
91+
date_added: 2025-04-01
92+
model_description: DP 2024 Q4, 37head share fitting 120GPU pretrain
93+
num_parameters: 6637036
94+
packages:
95+
deepmd-kit: P0131

0 commit comments

Comments
 (0)