diff --git a/deepmd/dpmodel/utils/type_embed.py b/deepmd/dpmodel/utils/type_embed.py index 04cd56b22..04c05b6a3 100644 --- a/deepmd/dpmodel/utils/type_embed.py +++ b/deepmd/dpmodel/utils/type_embed.py @@ -242,6 +242,5 @@ def get_econf_tebd(type_map, precision: str = "default"): [electronic_configuration_embedding[kk] for kk in type_map], dtype=PRECISION_DICT[precision], ) - econf_tebd /= econf_tebd.sum(-1, keepdims=True) # do normalization embed_input_dim = ECONF_DIM return econf_tebd, embed_input_dim