sync descriptor alias (#3374)

Add `se_a` to pt and dpmodel and add `dpa1` to tf. Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Mar 1, 2024 · 54b14ef · 54b14ef
1 parent 16c6db6
commit 54b14ef
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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
@@ -71,6 +71,7 @@
 log = logging.getLogger(__name__)
 
 
+@Descriptor.register("dpa1")
 @Descriptor.register("se_atten")
 class DescrptSeAtten(DescrptSeA):
     r"""Smooth version descriptor with attention.