fix unused var

deepmodeling · Jan 9, 2024 · f67fa21 · f67fa21
1 parent e0f077e
commit f67fa21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd_utils/model_format/se_e2_a.py b/deepmd_utils/model_format/se_e2_a.py
@@ -73,7 +73,7 @@ def __init__(
         for ii in range(self.ntypes):
             self.embeddings.append(
                 EmbeddingNet(
-                    1,
+                    in_dim,
                     self.neuron,
                     self.activation_function,
                     self.resnet_dt,