Skip to content

Commit 1d77808

Browse files
refine docstring
1 parent 9bddcdb commit 1d77808

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

deepmd/utils/argcheck.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3168,7 +3168,7 @@ def training_args(
31683168
doc_kf_blocksize = "The blocksize for the Kalman filter."
31693169
doc_model_prob = "The visiting probability of each model for each training step in the multi-task mode."
31703170
doc_data_dict = "The multiple definition of the data, used in the multi-task mode."
3171-
doc_acc_freq = "The accumulation steps for the gradients."
3171+
doc_acc_freq = "Gradient accumulation steps (number of steps to accumulate gradients before performing an update)."
31723172

31733173
arg_training_data = training_data_args()
31743174
arg_validation_data = validation_data_args()
@@ -3275,6 +3275,7 @@ def training_args(
32753275
"acc_freq",
32763276
int,
32773277
optional=True,
3278+
default=1,
32783279
doc=doc_only_pd_supported + doc_acc_freq,
32793280
),
32803281
]

0 commit comments

Comments
 (0)