change apply_bias doc

deepmodeling · Dec 15, 2024 · 5d3c50b · 5d3c50b
1 parent 2859f02
commit 5d3c50b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/deepmd/dpmodel/atomic_model/property_atomic_model.py b/deepmd/dpmodel/atomic_model/property_atomic_model.py
@@ -23,7 +23,8 @@ def apply_out_stat(
         ret: dict[str, np.ndarray],
         atype: np.ndarray,
     ):
-        """Apply the stat to each atomic output. In property fitting, each output will be multiplied by label std and then plus the label average value.
+        """Apply the stat to each atomic output.
+        In property fitting, each output will be multiplied by label std and then plus the label average value.
 
         Parameters
         ----------

diff --git a/deepmd/pt/model/atomic_model/property_atomic_model.py b/deepmd/pt/model/atomic_model/property_atomic_model.py
@@ -25,14 +25,14 @@ def apply_out_stat(
         atype: torch.Tensor,
     ):
         """Apply the stat to each atomic output.
-        This function defines how the bias is applied to the atomic output of the model.
+        In property fitting, each output will be multiplied by label std and then plus the label average value.
 
         Parameters
         ----------
         ret
             The returned dict by the forward_atomic method
         atype
-            The atom types. nf x nloc
+            The atom types. nf x nloc. It is useless in property fitting.
 
         """
         out_bias, out_std = self._fetch_out_stat(self.bias_keys)