File size: 1,735 Bytes
0df10a0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | {
"@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
}
}
|