Hugging Face's logo

deepmodelingcommunity
/
DPA3-Omol-Large

Model card Files
xet
 Community
penganyang commited on
Commit
a1075fb
·
verified ·
1 Parent(s): c1c3d42

Upload 3 files

Browse files
Files changed (3) hide show
  1. DPA3-Omol-Large.pt +3 -0
  2. input.json +0 -0
  3. input_finetune.json +215 -0
DPA3-Omol-Large.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc4d252b31450b41eb3546cc48f640ad0831c0b5d069ce27d996e0ff58fc037a
3
+ size 224057430
input.json ADDED
The diff for this file is too large to render. See raw diff
 
input_finetune.json ADDED
@@ -0,0 +1,215 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_comment": "that's all",
3
+ "model": {
4
+ "type_map": [
5
+ "H",
6
+ "He",
7
+ "Li",
8
+ "Be",
9
+ "B",
10
+ "C",
11
+ "N",
12
+ "O",
13
+ "F",
14
+ "Ne",
15
+ "Na",
16
+ "Mg",
17
+ "Al",
18
+ "Si",
19
+ "P",
20
+ "S",
21
+ "Cl",
22
+ "Ar",
23
+ "K",
24
+ "Ca",
25
+ "Sc",
26
+ "Ti",
27
+ "V",
28
+ "Cr",
29
+ "Mn",
30
+ "Fe",
31
+ "Co",
32
+ "Ni",
33
+ "Cu",
34
+ "Zn",
35
+ "Ga",
36
+ "Ge",
37
+ "As",
38
+ "Se",
39
+ "Br",
40
+ "Kr",
41
+ "Rb",
42
+ "Sr",
43
+ "Y",
44
+ "Zr",
45
+ "Nb",
46
+ "Mo",
47
+ "Tc",
48
+ "Ru",
49
+ "Rh",
50
+ "Pd",
51
+ "Ag",
52
+ "Cd",
53
+ "In",
54
+ "Sn",
55
+ "Sb",
56
+ "Te",
57
+ "I",
58
+ "Xe",
59
+ "Cs",
60
+ "Ba",
61
+ "La",
62
+ "Ce",
63
+ "Pr",
64
+ "Nd",
65
+ "Pm",
66
+ "Sm",
67
+ "Eu",
68
+ "Gd",
69
+ "Tb",
70
+ "Dy",
71
+ "Ho",
72
+ "Er",
73
+ "Tm",
74
+ "Yb",
75
+ "Lu",
76
+ "Hf",
77
+ "Ta",
78
+ "W",
79
+ "Re",
80
+ "Os",
81
+ "Ir",
82
+ "Pt",
83
+ "Au",
84
+ "Hg",
85
+ "Tl",
86
+ "Pb",
87
+ "Bi",
88
+ "Po",
89
+ "At",
90
+ "Rn",
91
+ "Fr",
92
+ "Ra",
93
+ "Ac",
94
+ "Th",
95
+ "Pa",
96
+ "U",
97
+ "Np",
98
+ "Pu",
99
+ "Am",
100
+ "Cm",
101
+ "Bk",
102
+ "Cf",
103
+ "Es",
104
+ "Fm",
105
+ "Md",
106
+ "No",
107
+ "Lr",
108
+ "Rf",
109
+ "Db",
110
+ "Sg",
111
+ "Bh",
112
+ "Hs",
113
+ "Mt",
114
+ "Ds",
115
+ "Rg",
116
+ "Cn",
117
+ "Nh",
118
+ "Fl",
119
+ "Mc",
120
+ "Lv",
121
+ "Ts",
122
+ "Og"
123
+ ],
124
+ "descriptor": {
125
+ "type": "dpa3",
126
+ "repflow": {
127
+ "n_dim": 256,
128
+ "e_dim": 256,
129
+ "a_dim": 256,
130
+ "nlayers": 12,
131
+ "e_rcut": 6.0,
132
+ "e_rcut_smth": 5.3,
133
+ "e_sel": 30,
134
+ "a_rcut": 4.5,
135
+ "a_rcut_smth": 4.0,
136
+ "a_sel": 15,
137
+ "axis_neuron": 4,
138
+ "fix_stat_std": 0.3,
139
+ "a_compress_rate": 1,
140
+ "a_compress_e_rate": 2,
141
+ "a_compress_use_split": true,
142
+ "update_angle": true,
143
+ "smooth_edge_update": true,
144
+ "edge_init_use_dist": true,
145
+ "use_dynamic_sel": false,
146
+ "sel_reduce_factor": 10.0,
147
+ "use_exp_switch": true,
148
+ "update_style": "res_residual",
149
+ "update_residual": 0.1,
150
+ "update_residual_init": "const"
151
+ },
152
+ "activation_function": "silut:10.0",
153
+ "use_tebd_bias": false,
154
+ "add_chg_spin_ebd": true,
155
+ "precision": "float32",
156
+ "concat_output_tebd": false
157
+ },
158
+ "fitting_net": {
159
+ "neuron": [
160
+ 240,
161
+ 240,
162
+ 240
163
+ ],
164
+ "numb_fparam": 2,
165
+ "default_fparam": [0.0, 1.0],
166
+ "resnet_dt": true,
167
+ "seed": 1,
168
+ "precision": "float32",
169
+ "activation_function": "silut:10.0",
170
+ "_comment": " that's all"
171
+ },
172
+ "_comment": " that's all"
173
+ },
174
+ "learning_rate": {
175
+ "type": "cosine",
176
+ "start_lr": 0.0005,
177
+ "stop_lr": 5e-06,
178
+ "_comment": "that's all"
179
+ },
180
+ "loss": {
181
+ "type": "ener",
182
+ "start_pref_e": 0.02,
183
+ "limit_pref_e": 20,
184
+ "start_pref_f": 100,
185
+ "limit_pref_f": 20,
186
+ "start_pref_v": 0,
187
+ "limit_pref_v": 0,
188
+ "_comment": " that's all"
189
+ },
190
+ "training": {
191
+ "stat_file": "../stat_file/OMol25_with_fparam_0_1.hdf5",
192
+ "training_data": {
193
+ "systems": [
194
+ ],
195
+ "batch_size": "auto:2048",
196
+ "_comment": "that's all"
197
+ },
198
+ "validation_data": {
199
+ "systems": [
200
+ ],
201
+ "batch_size": 1,
202
+ "_comment": "that's all"
203
+ },
204
+ "numb_steps": 2000000,
205
+ "warmup_steps": 0,
206
+ "gradient_max_norm": 5.0,
207
+ "seed": 10,
208
+ "opt_type": "AdamW",
209
+ "max_ckpt_keep": 10000000,
210
+ "disp_file": "lcurve.out",
211
+ "disp_freq": 100,
212
+ "save_freq": 10000,
213
+ "_comment": "that's all"
214
+ }
215
+ }