| { |
| "@class": "M3GNet", |
| "@module": "matgl.models._m3gnet", |
| "@model_version": 2, |
| "metadata": null, |
| "kwargs": { |
| "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": true, |
| "readout_type": "set2set", |
| "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 |
| } |
| } |
