Hugging Face's logo

deepmodelingcommunity
/
DPA

Model card Files
xet
Community
caic99 commited on
Commit
cac67b9
·
verified ·
1 Parent(s): 50dcaf9

Upload input files

Browse files
Files changed (2) hide show
  1. input_finetune.json +99 -0
  2. input_pretrain.json +347 -0
input_finetune.json ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_comment": "that's all",
3
+ "model": {
4
+ "type_map": [
5
+ "O",
6
+ "H"
7
+ ],
8
+ "descriptor": {
9
+ "type": "dpa3",
10
+ "repflow": {
11
+ "n_dim": 128,
12
+ "e_dim": 64,
13
+ "a_dim": 32,
14
+ "nlayers": 16,
15
+ "e_rcut": 6.0,
16
+ "e_rcut_smth": 5.3,
17
+ "e_sel": 1200,
18
+ "a_rcut": 4.0,
19
+ "a_rcut_smth": 3.5,
20
+ "a_sel": 300,
21
+ "axis_neuron": 4,
22
+ "fix_stat_std": 0.3,
23
+ "a_compress_rate": 1,
24
+ "a_compress_e_rate": 2,
25
+ "a_compress_use_split": true,
26
+ "update_angle": true,
27
+ "smooth_edge_update": true,
28
+ "use_dynamic_sel": true,
29
+ "sel_reduce_factor": 10.0,
30
+ "use_exp_switch": true,
31
+ "update_style": "res_residual",
32
+ "update_residual": 0.1,
33
+ "update_residual_init": "const"
34
+ },
35
+ "activation_function": "silut:3.0",
36
+ "use_tebd_bias": false,
37
+ "precision": "float32",
38
+ "concat_output_tebd": false
39
+ },
40
+ "fitting_net": {
41
+ "neuron": [
42
+ 240,
43
+ 240,
44
+ 240
45
+ ],
46
+ "dim_case_embd": 31,
47
+ "resnet_dt": true,
48
+ "precision": "float32",
49
+ "activation_function": "silut:3.0",
50
+ "seed": 1,
51
+ "_comment": " that's all"
52
+ },
53
+ "_comment": " that's all"
54
+ },
55
+ "learning_rate": {
56
+ "type": "exp",
57
+ "decay_steps": 5000,
58
+ "start_lr": 0.001,
59
+ "stop_lr": 3e-5,
60
+ "_comment": "that's all"
61
+ },
62
+ "loss": {
63
+ "type": "ener",
64
+ "start_pref_e": 0.2,
65
+ "limit_pref_e": 20,
66
+ "start_pref_f": 100,
67
+ "limit_pref_f": 60,
68
+ "start_pref_v": 0.02,
69
+ "limit_pref_v": 1,
70
+ "_comment": " that's all"
71
+ },
72
+ "training": {
73
+ "stat_file": "./dpa3.hdf5",
74
+ "training_data": {
75
+ "systems": [
76
+ "../data/data_0",
77
+ "../data/data_1",
78
+ "../data/data_2"
79
+ ],
80
+ "batch_size": 1,
81
+ "_comment": "that's all"
82
+ },
83
+ "validation_data": {
84
+ "systems": [
85
+ "../data/data_3"
86
+ ],
87
+ "batch_size": 1,
88
+ "_comment": "that's all"
89
+ },
90
+ "numb_steps": 1000000,
91
+ "warmup_steps": 0,
92
+ "gradient_max_norm": 5.0,
93
+ "seed": 10,
94
+ "disp_file": "lcurve.out",
95
+ "disp_freq": 100,
96
+ "save_freq": 2000,
97
+ "_comment": "that's all"
98
+ }
99
+ }
input_pretrain.json ADDED
@@ -0,0 +1,347 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_comment": "that's all",
3
+ "model": {
4
+ "shared_dict": {
5
+ "type_map_all": [
6
+ "H",
7
+ "He",
8
+ "Li",
9
+ "Be",
10
+ "B",
11
+ "C",
12
+ "N",
13
+ "O",
14
+ "F",
15
+ "Ne",
16
+ "Na",
17
+ "Mg",
18
+ "Al",
19
+ "Si",
20
+ "P",
21
+ "S",
22
+ "Cl",
23
+ "Ar",
24
+ "K",
25
+ "Ca",
26
+ "Sc",
27
+ "Ti",
28
+ "V",
29
+ "Cr",
30
+ "Mn",
31
+ "Fe",
32
+ "Co",
33
+ "Ni",
34
+ "Cu",
35
+ "Zn",
36
+ "Ga",
37
+ "Ge",
38
+ "As",
39
+ "Se",
40
+ "Br",
41
+ "Kr",
42
+ "Rb",
43
+ "Sr",
44
+ "Y",
45
+ "Zr",
46
+ "Nb",
47
+ "Mo",
48
+ "Tc",
49
+ "Ru",
50
+ "Rh",
51
+ "Pd",
52
+ "Ag",
53
+ "Cd",
54
+ "In",
55
+ "Sn",
56
+ "Sb",
57
+ "Te",
58
+ "I",
59
+ "Xe",
60
+ "Cs",
61
+ "Ba",
62
+ "La",
63
+ "Ce",
64
+ "Pr",
65
+ "Nd",
66
+ "Pm",
67
+ "Sm",
68
+ "Eu",
69
+ "Gd",
70
+ "Tb",
71
+ "Dy",
72
+ "Ho",
73
+ "Er",
74
+ "Tm",
75
+ "Yb",
76
+ "Lu",
77
+ "Hf",
78
+ "Ta",
79
+ "W",
80
+ "Re",
81
+ "Os",
82
+ "Ir",
83
+ "Pt",
84
+ "Au",
85
+ "Hg",
86
+ "Tl",
87
+ "Pb",
88
+ "Bi",
89
+ "Po",
90
+ "At",
91
+ "Rn",
92
+ "Fr",
93
+ "Ra",
94
+ "Ac",
95
+ "Th",
96
+ "Pa",
97
+ "U",
98
+ "Np",
99
+ "Pu",
100
+ "Am",
101
+ "Cm",
102
+ "Bk",
103
+ "Cf",
104
+ "Es",
105
+ "Fm",
106
+ "Md",
107
+ "No",
108
+ "Lr",
109
+ "Rf",
110
+ "Db",
111
+ "Sg",
112
+ "Bh",
113
+ "Hs",
114
+ "Mt",
115
+ "Ds",
116
+ "Rg",
117
+ "Cn",
118
+ "Nh",
119
+ "Fl",
120
+ "Mc",
121
+ "Lv",
122
+ "Ts",
123
+ "Og"
124
+ ],
125
+ "dpa3_descriptor": {
126
+ "type": "dpa3",
127
+ "repflow": {
128
+ "n_dim": 128,
129
+ "e_dim": 64,
130
+ "a_dim": 32,
131
+ "nlayers": 16,
132
+ "e_rcut": 6.0,
133
+ "e_rcut_smth": 5.3,
134
+ "e_sel": 1200,
135
+ "a_rcut": 4.0,
136
+ "a_rcut_smth": 3.5,
137
+ "a_sel": 300,
138
+ "axis_neuron": 4,
139
+ "skip_stat": true,
140
+ "a_compress_rate": 1,
141
+ "a_compress_e_rate": 2,
142
+ "a_compress_use_split": true,
143
+ "update_angle": true,
144
+ "smooth_edge_update": true,
145
+ "use_dynamic_sel": true,
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": "custom_silu:3.0",
153
+ "precision": "float32",
154
+ "use_tebd_bias": false,
155
+ "concat_output_tebd": false
156
+ },
157
+ "shared_fitting": {
158
+ "neuron": [
159
+ 240,
160
+ 240,
161
+ 240
162
+ ],
163
+ "activation_function": "custom_silu:3.0",
164
+ "resnet_dt": true,
165
+ "precision": "float32",
166
+ "dim_case_embd": 31,
167
+ "seed": 1,
168
+ "_comment": " that's all"
169
+ },
170
+ "_comment": "that's all"
171
+ },
172
+ "model_dict": {
173
+ "Organic_Reactions": {
174
+ "type_map": "type_map_all",
175
+ "descriptor": "dpa3_descriptor",
176
+ "fitting_net": "shared_fitting"
177
+ }
178
+ }
179
+ },
180
+ "learning_rate": {
181
+ "type": "exp",
182
+ "decay_steps": 5000,
183
+ "start_lr": 0.001,
184
+ "stop_lr": 1e-06,
185
+ "_comment": "that's all"
186
+ },
187
+ "loss_dict": {
188
+ "Organic_Reactions": {
189
+ "type": "ener",
190
+ "start_pref_e": 0.2,
191
+ "limit_pref_e": 20,
192
+ "start_pref_f": 100,
193
+ "limit_pref_f": 60,
194
+ "start_pref_v": 0.02,
195
+ "limit_pref_v": 1
196
+ }
197
+ },
198
+ "training": {
199
+ "model_prob": {
200
+ "Organic_Reactions": 11.93
201
+ },
202
+ "data_dict": {
203
+ "Organic_Reactions": {
204
+ "stat_file": "/mnt/data_nas/zhangd/workplace/stat_files/multitask/beta4/Organic_Reactions.hdf5",
205
+ "training_data": {
206
+ "systems": [
207
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/10/train",
208
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/11/train",
209
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/12/train",
210
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/13/train",
211
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/14/train",
212
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/15/train",
213
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/16/train",
214
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/17/train",
215
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/18/train",
216
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/19/train",
217
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/20/train",
218
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/21/train",
219
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/22/train",
220
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/23/train",
221
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/24/train",
222
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/25/train",
223
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/26/train",
224
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/27/train",
225
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/28/train",
226
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/29/train",
227
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/30/train",
228
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/31/train",
229
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/32/train",
230
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/33/train",
231
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/4/train",
232
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/5/train",
233
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/6/train",
234
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/7/train",
235
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/8/train",
236
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/9/train",
237
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/10/train",
238
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/11/train",
239
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/12/train",
240
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/13/train",
241
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/14/train",
242
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/15/train",
243
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/16/train",
244
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/17/train",
245
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/18/train",
246
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/19/train",
247
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/20/train",
248
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/21/train",
249
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/22/train",
250
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/23/train",
251
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/24/train",
252
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/25/train",
253
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/26/train",
254
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/27/train",
255
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/28/train",
256
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/29/train",
257
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/30/train",
258
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/31/train",
259
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/32/train",
260
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/33/train",
261
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/5/train",
262
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/6/train",
263
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/7/train",
264
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/8/train",
265
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/9/train"
266
+ ],
267
+ "batch_size": "auto:128",
268
+ "_comment": "that's all"
269
+ },
270
+ "validation_data": {
271
+ "systems": [
272
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/10/valid",
273
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/11/valid",
274
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/12/valid",
275
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/13/valid",
276
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/14/valid",
277
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/15/valid",
278
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/16/valid",
279
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/17/valid",
280
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/18/valid",
281
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/19/valid",
282
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/20/valid",
283
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/21/valid",
284
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/22/valid",
285
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/23/valid",
286
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/24/valid",
287
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/25/valid",
288
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/26/valid",
289
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/27/valid",
290
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/28/valid",
291
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/29/valid",
292
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/30/valid",
293
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/31/valid",
294
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/32/valid",
295
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/33/valid",
296
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/4/valid",
297
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/5/valid",
298
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/6/valid",
299
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/7/valid",
300
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/8/valid",
301
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/77k_rxn_xtb2/9/valid",
302
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/10/valid",
303
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/11/valid",
304
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/12/valid",
305
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/13/valid",
306
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/14/valid",
307
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/15/valid",
308
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/16/valid",
309
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/17/valid",
310
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/18/valid",
311
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/19/valid",
312
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/20/valid",
313
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/21/valid",
314
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/22/valid",
315
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/23/valid",
316
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/24/valid",
317
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/25/valid",
318
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/26/valid",
319
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/27/valid",
320
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/28/valid",
321
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/29/valid",
322
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/30/valid",
323
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/31/valid",
324
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/32/valid",
325
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/33/valid",
326
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/5/valid",
327
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/6/valid",
328
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/7/valid",
329
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/8/valid",
330
+ "/mnt/data_nas/public/public_training_data/organic_reactions_clean_split/80k_rxn_xtb2/9/valid"
331
+ ],
332
+ "batch_size": 1,
333
+ "_comment": "that's all"
334
+ }
335
+ }
336
+ },
337
+ "numb_steps": 8000000,
338
+ "gradient_max_norm": 5.0,
339
+ "max_ckpt_keep": 5000,
340
+ "warmup_steps": 0,
341
+ "seed": 10,
342
+ "disp_file": "lcurve.out",
343
+ "disp_freq": 10000,
344
+ "save_freq": 10000,
345
+ "_comment": "that's all"
346
+ }
347
+ }