fix se_atten variable names when suffix is given

It seems that deepmodeling#1891 wrongly added a suffix to the variable names to make them like `filter_type_all_suffix/matrix_1_suffix`. However, it is expected to be `filter_type_all_suffix/matrix_1` which is consistent in other classes and methods. Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Jun 23, 2023 · 35797eb · 35797eb
1 parent 473893e
commit 35797eb
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/deepmd/descriptor/se_atten.py b/deepmd/descriptor/se_atten.py
@@ -649,7 +649,6 @@ def _pass_filter(
             type_i,
             natoms,
             name="filter_type_all" + suffix,
-            suffix=suffix,
             reuse=reuse,
             trainable=trainable,
             activation_fn=self.filter_activation_fn,