File size: 1,810 Bytes
9a1e148 | 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 56 | {
"@class": "Potential",
"@module": "matgl.apps.pes",
"@model_version": 3,
"metadata": null,
"kwargs": {
"model": {
"@class": "SO3Net",
"@module": "matgl.models._so3net",
"@model_version": 0,
"init_args": {
"element_types": [
"H",
"C",
"N",
"O"
],
"units": 64,
"dim_state_embedding": 0,
"ntypes_state": null,
"dim_state_feats": null,
"nblocks": 3,
"nmax": 5,
"cutoff": 5.0,
"rbf_learnable": false,
"target_property": "atomwise",
"task_type": "regression",
"readout_type": "weighted_atom",
"niters_set2set": 3,
"nlayers_set2set": 3,
"nlayers_readout": 2,
"is_intensive": false,
"include_state": false,
"use_vector_representation": false,
"correct_charges": false,
"predict_dipole_magnitude": false,
"activation_type": "swish",
"ntargets": 1,
"return_vector_representation": false,
"dim_node_embedding": 64,
"lmax": 2
}
},
"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
}
}
|