File size: 1,868 Bytes
52c8168 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | {
"model": {
"type_map": [
"O",
"H"
],
"descriptor": {
"type": "dpa1",
"sel": 120,
"rcut_smth": 0.5,
"rcut": 6.0,
"neuron": [
25,
50,
100
],
"tebd_dim": 8,
"axis_neuron": 16,
"type_one_side": true,
"attn": 128,
"attn_layer": 2,
"attn_dotr": true,
"attn_mask": false,
"activation_function": "tanh",
"scaling_factor": 1.0,
"normalize": true,
"temperature": 1.0
},
"fitting_net": {
"neuron": [
240,
240,
240
],
"resnet_dt": true,
"seed": 1,
"_comment": " that's all"
},
"data_stat_nbatch": 20,
"_comment": " that's all"
},
"learning_rate": {
"type": "exp",
"decay_steps": 5000,
"start_lr": 0.001,
"stop_lr": 3.51e-8,
"_comment": "that's all"
},
"loss": {
"type": "ener",
"start_pref_e": 0.02,
"limit_pref_e": 1,
"start_pref_f": 1000,
"limit_pref_f": 1,
"_comment": " that's all"
},
"training": {
"stat_file": "./dpa1.hdf5",
"training_data": {
"systems": [
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/dp/water/data_0",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/dp/water/data_1",
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/dp/water/data_2"
],
"batch_size": 4,
"_comment": "that's all"
},
"validation_data": {
"systems": [
"/public/share/sugonhpcapp01/onestore/onedatasets/matchem/dp/water/data_3"
],
"batch_size": 4,
"numb_btch": 3,
"_comment": "that's all"
},
"numb_steps": 1000,
"seed": 10,
"disp_file": "lcurve.out",
"disp_freq": 100,
"save_freq": 1000,
"_comment": "that's all"
},
"_comment": "that's all"
}
|