diff --git a/deepmd/dpmodel/descriptor/se_e2_a.py b/deepmd/dpmodel/descriptor/se_e2_a.py index 5e72653f1d..72ab99f755 100644 --- a/deepmd/dpmodel/descriptor/se_e2_a.py +++ b/deepmd/dpmodel/descriptor/se_e2_a.py @@ -46,6 +46,7 @@ @BaseDescriptor.register("se_e2_a") +@BaseDescriptor.register("se_a") class DescrptSeA(NativeOP, BaseDescriptor): r"""DeepPot-SE constructed from all information (both angular and radial) of atomic configurations. The embedding takes the distance between atoms as input. diff --git a/deepmd/pt/model/descriptor/se_a.py b/deepmd/pt/model/descriptor/se_a.py index fc2cf60531..3c8d1e325b 100644 --- a/deepmd/pt/model/descriptor/se_a.py +++ b/deepmd/pt/model/descriptor/se_a.py @@ -63,6 +63,7 @@ @BaseDescriptor.register("se_e2_a") +@BaseDescriptor.register("se_a") class DescrptSeA(BaseDescriptor, torch.nn.Module): def __init__( self, diff --git a/deepmd/tf/descriptor/se_atten.py b/deepmd/tf/descriptor/se_atten.py index 4be5cbd164..8d80c10ba5 100644 --- a/deepmd/tf/descriptor/se_atten.py +++ b/deepmd/tf/descriptor/se_atten.py @@ -71,6 +71,7 @@ log = logging.getLogger(__name__) +@Descriptor.register("dpa1") @Descriptor.register("se_atten") class DescrptSeAtten(DescrptSeA): r"""Smooth version descriptor with attention.