| { | |
| "@class": "Potential", | |
| "@module": "matgl.apps.pes", | |
| "@model_version": 3, | |
| "metadata": null, | |
| "kwargs": { | |
| "model": { | |
| "@class": "TensorNet", | |
| "@module": "matgl.models._tensornet_dgl", | |
| "@model_version": 1, | |
| "init_args": { | |
| "element_types": [ | |
| "H", | |
| "C", | |
| "N", | |
| "O" | |
| ], | |
| "ntypes_state": null, | |
| "dim_state_embedding": 0, | |
| "dim_state_feats": null, | |
| "include_state": false, | |
| "max_n": 3, | |
| "max_l": 3, | |
| "rbf_type": "SphericalBessel", | |
| "use_smooth": true, | |
| "activation_type": "swish", | |
| "width": 0.5, | |
| "readout_type": "weighted_atom", | |
| "task_type": "regression", | |
| "niters_set2set": 3, | |
| "nlayers_set2set": 3, | |
| "field": "node_feat", | |
| "is_intensive": false, | |
| "ntargets": 1, | |
| "cutoff": 5.0, | |
| "dtype": "torch.float32", | |
| "equivariance_invariance_group": "O(3)", | |
| "nblocks": 2, | |
| "num_rbf": 3, | |
| "units": 64 | |
| } | |
| }, | |
| "data_mean": "tensor(0.)", | |
| "data_std": "tensor(3.70301199)", | |
| "element_refs": "tensor([ -16.51807976, -1036.37304688, -1488.65246582, -2046.00830078])", | |
| "calc_forces": true, | |
| "calc_stresses": false, | |
| "calc_hessian": false, | |
| "calc_magmom": false, | |
| "calc_repuls": false, | |
| "zbl_trainable": false, | |
| "debug_mode": false | |
| } | |
| } | |