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