| { |
| "_comment": "that's all", |
| "model": { |
| "type_map": [ |
| "H", |
| "He", |
| "Li", |
| "Be", |
| "B", |
| "C", |
| "N", |
| "O", |
| "F", |
| "Ne", |
| "Na", |
| "Mg", |
| "Al", |
| "Si", |
| "P", |
| "S", |
| "Cl", |
| "Ar", |
| "K", |
| "Ca", |
| "Sc", |
| "Ti", |
| "V", |
| "Cr", |
| "Mn", |
| "Fe", |
| "Co", |
| "Ni", |
| "Cu", |
| "Zn", |
| "Ga", |
| "Ge", |
| "As", |
| "Se", |
| "Br", |
| "Kr", |
| "Rb", |
| "Sr", |
| "Y", |
| "Zr", |
| "Nb", |
| "Mo", |
| "Tc", |
| "Ru", |
| "Rh", |
| "Pd", |
| "Ag", |
| "Cd", |
| "In", |
| "Sn", |
| "Sb", |
| "Te", |
| "I", |
| "Xe", |
| "Cs", |
| "Ba", |
| "La", |
| "Ce", |
| "Pr", |
| "Nd", |
| "Pm", |
| "Sm", |
| "Eu", |
| "Gd", |
| "Tb", |
| "Dy", |
| "Ho", |
| "Er", |
| "Tm", |
| "Yb", |
| "Lu", |
| "Hf", |
| "Ta", |
| "W", |
| "Re", |
| "Os", |
| "Ir", |
| "Pt", |
| "Au", |
| "Hg", |
| "Tl", |
| "Pb", |
| "Bi", |
| "Po", |
| "At", |
| "Rn", |
| "Fr", |
| "Ra", |
| "Ac", |
| "Th", |
| "Pa", |
| "U", |
| "Np", |
| "Pu", |
| "Am", |
| "Cm", |
| "Bk", |
| "Cf", |
| "Es", |
| "Fm", |
| "Md", |
| "No", |
| "Lr", |
| "Rf", |
| "Db", |
| "Sg", |
| "Bh", |
| "Hs", |
| "Mt", |
| "Ds", |
| "Rg", |
| "Cn", |
| "Nh", |
| "Fl", |
| "Mc", |
| "Lv", |
| "Ts", |
| "Og" |
| ], |
| "descriptor": { |
| "type": "dpa3", |
| "repflow": { |
| "n_dim": 256, |
| "e_dim": 256, |
| "a_dim": 256, |
| "nlayers": 12, |
| "e_rcut": 6.0, |
| "e_rcut_smth": 5.3, |
| "e_sel": 30, |
| "a_rcut": 4.5, |
| "a_rcut_smth": 4.0, |
| "a_sel": 15, |
| "axis_neuron": 4, |
| "fix_stat_std": 0.3, |
| "a_compress_rate": 1, |
| "a_compress_e_rate": 2, |
| "a_compress_use_split": true, |
| "update_angle": true, |
| "smooth_edge_update": true, |
| "edge_init_use_dist": true, |
| "use_dynamic_sel": false, |
| "sel_reduce_factor": 10.0, |
| "use_exp_switch": true, |
| "update_style": "res_residual", |
| "update_residual": 0.1, |
| "update_residual_init": "const" |
| }, |
| "activation_function": "silut:10.0", |
| "use_tebd_bias": false, |
| "add_chg_spin_ebd": true, |
| "precision": "float32", |
| "concat_output_tebd": false |
| }, |
| "fitting_net": { |
| "neuron": [ |
| 240, |
| 240, |
| 240 |
| ], |
| "numb_fparam": 2, |
| "default_fparam": [0.0, 1.0], |
| "resnet_dt": true, |
| "seed": 1, |
| "precision": "float32", |
| "activation_function": "silut:10.0", |
| "_comment": " that's all" |
| }, |
| "_comment": " that's all" |
| }, |
| "learning_rate": { |
| "type": "cosine", |
| "start_lr": 0.0005, |
| "stop_lr": 5e-06, |
| "_comment": "that's all" |
| }, |
| "loss": { |
| "type": "ener", |
| "start_pref_e": 0.02, |
| "limit_pref_e": 20, |
| "start_pref_f": 100, |
| "limit_pref_f": 20, |
| "start_pref_v": 0, |
| "limit_pref_v": 0, |
| "_comment": " that's all" |
| }, |
| "training": { |
| "stat_file": "../stat_file/OMol25_with_fparam_0_1.hdf5", |
| "training_data": { |
| "systems": [ |
| ], |
| "batch_size": "auto:2048", |
| "_comment": "that's all" |
| }, |
| "validation_data": { |
| "systems": [ |
| ], |
| "batch_size": 1, |
| "_comment": "that's all" |
| }, |
| "numb_steps": 2000000, |
| "warmup_steps": 0, |
| "gradient_max_norm": 5.0, |
| "seed": 10, |
| "opt_type": "AdamW", |
| "max_ckpt_keep": 10000000, |
| "disp_file": "lcurve.out", |
| "disp_freq": 100, |
| "save_freq": 10000, |
| "_comment": "that's all" |
| } |
| } |
|
|
