From fec6462262944e6d3bdf0a441bf581fbe0c95167 Mon Sep 17 00:00:00 2001 From: Duo <50307526+iProzd@users.noreply.github.com> Date: Fri, 3 Jan 2025 14:53:09 +0800 Subject: [PATCH] fix typo --- deepmd/dpmodel/descriptor/dpa3.py | 2 +- deepmd/pt/model/descriptor/repflows.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/deepmd/dpmodel/descriptor/dpa3.py b/deepmd/dpmodel/descriptor/dpa3.py index d8fe4f1c24..356743e565 100644 --- 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 index 143a2d1f85..a6e76333c3 100644 --- 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.