Skip to content

Commit 8acae00

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 56fabc9 commit 8acae00

1 file changed

Lines changed: 103 additions & 103 deletions

File tree

examples/density/dpa3/input.json

Lines changed: 103 additions & 103 deletions
Original file line numberDiff line numberDiff line change
@@ -1,107 +1,107 @@
11
{
2-
"_comment": "that's all",
3-
"model": {
4-
"type_map": [
5-
"Li",
6-
"Ni",
7-
"Co",
8-
"Mn",
9-
"O",
10-
"C",
11-
"H",
12-
"N",
13-
"F",
14-
"X"
15-
],
16-
"descriptor": {
17-
"type": "dpa3",
18-
"repflow": {
19-
"n_dim": 128,
20-
"e_dim": 64,
21-
"a_dim": 32,
22-
"nlayers": 12,
23-
"e_rcut": 6.0,
24-
"e_rcut_smth": 5.3,
25-
"e_sel": 120,
26-
"a_rcut": 4.0,
27-
"a_rcut_smth": 3.5,
28-
"a_sel": 30,
29-
"axis_neuron": 4,
30-
"skip_stat": true,
31-
"a_compress_rate": 1,
32-
"a_compress_e_rate": 2,
33-
"a_compress_use_split": true,
34-
"update_angle": true,
35-
"smooth_edge_update": true,
36-
"use_dynamic_sel": true,
37-
"sel_reduce_factor": 10.0,
38-
"edge_use_dist": true,
39-
"use_env_envelope": true,
40-
"optim_update": true,
41-
"update_style": "res_residual",
42-
"update_residual": 0.1,
43-
"update_residual_init": "const"
44-
},
45-
"activation_function": "custom_silu:10.0",
46-
"use_tebd_bias": false,
47-
"env_protection": 0.1,
48-
"precision": "float32",
49-
"concat_output_tebd": false
50-
},
51-
"fitting_net": {
52-
"type": "density",
53-
"neuron": [
54-
240,
55-
240,
56-
240
57-
],
58-
"resnet_dt": true,
59-
"seed": 1,
60-
"precision": "float32",
61-
"activation_function": "silu",
62-
"_comment": " that's all"
63-
},
64-
"_comment": " that's all"
2+
"_comment": "that's all",
3+
"model": {
4+
"type_map": [
5+
"Li",
6+
"Ni",
7+
"Co",
8+
"Mn",
9+
"O",
10+
"C",
11+
"H",
12+
"N",
13+
"F",
14+
"X"
15+
],
16+
"descriptor": {
17+
"type": "dpa3",
18+
"repflow": {
19+
"n_dim": 128,
20+
"e_dim": 64,
21+
"a_dim": 32,
22+
"nlayers": 12,
23+
"e_rcut": 6.0,
24+
"e_rcut_smth": 5.3,
25+
"e_sel": 120,
26+
"a_rcut": 4.0,
27+
"a_rcut_smth": 3.5,
28+
"a_sel": 30,
29+
"axis_neuron": 4,
30+
"skip_stat": true,
31+
"a_compress_rate": 1,
32+
"a_compress_e_rate": 2,
33+
"a_compress_use_split": true,
34+
"update_angle": true,
35+
"smooth_edge_update": true,
36+
"use_dynamic_sel": true,
37+
"sel_reduce_factor": 10.0,
38+
"edge_use_dist": true,
39+
"use_env_envelope": true,
40+
"optim_update": true,
41+
"update_style": "res_residual",
42+
"update_residual": 0.1,
43+
"update_residual_init": "const"
44+
},
45+
"activation_function": "custom_silu:10.0",
46+
"use_tebd_bias": false,
47+
"env_protection": 0.1,
48+
"precision": "float32",
49+
"concat_output_tebd": false
6550
},
66-
"learning_rate": {
67-
"type": "exp",
68-
"decay_steps": 5000,
69-
"start_lr": 0.001,
70-
"stop_lr": 1e-05,
71-
"_comment": "that's all"
51+
"fitting_net": {
52+
"type": "density",
53+
"neuron": [
54+
240,
55+
240,
56+
240
57+
],
58+
"resnet_dt": true,
59+
"seed": 1,
60+
"precision": "float32",
61+
"activation_function": "silu",
62+
"_comment": " that's all"
7263
},
73-
"loss": {
74-
"type": "grid_density",
75-
"start_pref_d": 1,
76-
"limit_pref_d": 1,
77-
"_comment": " that's all"
64+
"_comment": " that's all"
65+
},
66+
"learning_rate": {
67+
"type": "exp",
68+
"decay_steps": 5000,
69+
"start_lr": 0.001,
70+
"stop_lr": 1e-05,
71+
"_comment": "that's all"
72+
},
73+
"loss": {
74+
"type": "grid_density",
75+
"start_pref_d": 1,
76+
"limit_pref_d": 1,
77+
"_comment": " that's all"
78+
},
79+
"training": {
80+
"stat_file": "./qm9_charge_density.hdf5",
81+
"training_data": {
82+
"systems": [
83+
"../dataset/qm9/C7H15NO_train"
84+
],
85+
"batch_size": "auto:128",
86+
"_comment": "that's all"
7887
},
79-
"training": {
80-
"stat_file": "./qm9_charge_density.hdf5",
81-
"training_data": {
82-
"systems": [
83-
"../dataset/qm9/C7H15NO_train"
84-
],
85-
"batch_size": "auto:128",
86-
"_comment": "that's all"
87-
},
88-
"validation_data": {
89-
"systems": [
90-
"../dataset/qm9/C7H15NO_val"
91-
],
92-
"batch_size": 1,
93-
"numb_btch": 3,
94-
"_comment": "that's all"
95-
},
96-
"numb_steps": 100,
97-
"warmup_steps": 0,
98-
"gradient_max_norm": 5.0,
99-
"seed": 10,
100-
"max_ckpt_keep": 100,
101-
"opt_type": "AdamW",
102-
"disp_file": "lcurve.out",
103-
"disp_freq": 100,
104-
"save_freq": 10000,
105-
"_comment": "that's all"
106-
}
107-
}
88+
"validation_data": {
89+
"systems": [
90+
"../dataset/qm9/C7H15NO_val"
91+
],
92+
"batch_size": 1,
93+
"numb_btch": 3,
94+
"_comment": "that's all"
95+
},
96+
"numb_steps": 100,
97+
"warmup_steps": 0,
98+
"gradient_max_norm": 5.0,
99+
"seed": 10,
100+
"max_ckpt_keep": 100,
101+
"opt_type": "AdamW",
102+
"disp_file": "lcurve.out",
103+
"disp_freq": 100,
104+
"save_freq": 10000,
105+
"_comment": "that's all"
106+
}
107+
}

0 commit comments

Comments
 (0)