yiqingchen's picture
Update PyG model files and metrics
a68cd90 verified
{
"@class": "Potential",
"@module": "matgl.apps.pes",
"@model_version": 3,
"metadata": {
"dataset": "r2scan",
"backend": "PYG"
},
"kwargs": {
"model": {
"@class": "M3GNet",
"@module": "matgl.models",
"@model_version": 2,
"init_args": {
"element_types": [
"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",
"Ac",
"Th",
"Pa",
"U",
"Np",
"Pu"
],
"dim_node_embedding": 64,
"dim_edge_embedding": 64,
"dim_state_embedding": 0,
"ntypes_state": null,
"dim_state_feats": null,
"max_n": 3,
"max_l": 3,
"nblocks": 3,
"rbf_type": "SphericalBessel",
"is_intensive": false,
"readout_type": "weighted_atom",
"task_type": "regression",
"cutoff": 5.0,
"threebody_cutoff": 4.0,
"units": 64,
"ntargets": 1,
"use_smooth": false,
"use_phi": false,
"niters_set2set": 3,
"nlayers_set2set": 3,
"field": "node_feat",
"include_state": false,
"activation_type": "swish",
"dropout": null
}
},
"data_mean": 0.0,
"data_std": 1.704247236251831,
"element_refs": "[ -1.122561 -0.32636788 -0.77574867 -0.5550596 -1.1719562\n -2.3715327 -4.4272256 -3.220797 -2.2466795 -1.8758706\n -2.479163 -2.4600985 -3.136107 -4.104173 -5.5835047\n -5.1175885 -4.8080845 -4.8154817 -5.1619835 -4.9643736\n -7.3758383 -7.8934813 -9.399335 -11.586129 -11.621531\n -9.751238 -8.069623 -7.33886 -6.9637227 -7.4429703\n -8.615311 -10.0013895 -11.835241 -11.787121 -11.943188\n -12.531252 -13.168978 -13.307922 -16.17722 -16.686445\n -17.68513 -20.436321 -19.678934 -18.305193 -18.300009\n -18.779137 -18.478647 -19.04303 -20.181932 -21.364101\n -23.147707 -23.149889 -23.310696 -23.804207 -24.337032\n -25.72952 -25.777267 -26.708364 -28.031284 -30.121346\n -31.833288 -34.206905 -37.50571 -37.09642 -35.363342\n -35.18799 -34.49156 -33.455944 -32.874195 -33.50674\n -33.752407 -38.721516 -40.454357 -43.421467 -45.010036\n -44.95498 -45.217335 -45.783497 -47.1755 -48.910515\n -51.097305 -53.336815 -56.19985 -65.085655 -67.22815\n -70.10834 -74.35432 -78.24114 -82.93461 ]",
"calc_forces": true,
"calc_stresses": true,
"calc_hessian": false,
"calc_magmom": false,
"calc_charge": false,
"calc_repuls": false,
"zbl_trainable": false,
"debug_mode": false
}
}