fix typo

iProzd · Jan 3, 2025 · fec6462 · fec6462
1 parent fe5c318
commit fec6462
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deepmd/dpmodel/descriptor/dpa3.py b/deepmd/dpmodel/descriptor/dpa3.py
@@ -55,7 +55,7 @@ def __init__(
         a_compress_e_rate : int, optional
             The extra compression rate for edge in angular message compression. The default value is 1.
             When using angular message compression with a_compress_rate c and a_compress_e_rate c_e,
-            the edge dimension will be compressed to (c_e * e_dim / 2c) within the angular message.
+            the edge dimension will be compressed to (c_e * a_dim / 2c) within the angular message.
         a_compress_use_split : bool, optional
             Whether to split first sub-vectors instead of linear mapping during angular message compression.
             The default value is False.

diff --git a/deepmd/pt/model/descriptor/repflows.py b/deepmd/pt/model/descriptor/repflows.py
@@ -133,7 +133,7 @@ def __init__(
         a_compress_e_rate : int, optional
             The extra compression rate for edge in angular message compression. The default value is 1.
             When using angular message compression with a_compress_rate c and a_compress_e_rate c_e,
-            the edge dimension will be compressed to (c_e * e_dim / 2c) within the angular message.
+            the edge dimension will be compressed to (c_e * a_dim / 2c) within the angular message.
         a_compress_use_split : bool, optional
             Whether to split first sub-vectors instead of linear mapping during angular message compression.
             The default value is False.