add doc str

deepmodeling · Apr 6, 2024 · a3e6f57 · a3e6f57
1 parent 1cbbd9b
commit a3e6f57
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepmd/pt/model/atomic_model/base_atomic_model.py b/deepmd/pt/model/atomic_model/base_atomic_model.py
@@ -353,6 +353,8 @@ def change_out_bias(
             'change-by-statistic' : perform predictions on labels of target dataset,
                     and do least square on the errors to obtain the target shift as bias.
             'set-by-statistic' : directly use the statistic output bias in the target dataset.
+        stat_file_path : Optional[DPPath]
+            The path to the stat file.
         """
         if bias_adjust_mode == "change-by-statistic":
             delta_bias, out_std = compute_output_stats(