diff --git a/deepmd/dpmodel/descriptor/se_e2_a.py b/deepmd/dpmodel/descriptor/se_e2_a.py index f6b1c5677e..891f308edc 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 d836b48992..ee84da6bc2 100644 --- a/deepmd/pt/model/descriptor/se_a.py +++ b/deepmd/pt/model/descriptor/se_a.py @@ -65,6 +65,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.