{ "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" }