Skip to content

Commit d272b50

Browse files
add acc_freq doc
1 parent ed992ad commit d272b50

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

deepmd/utils/argcheck.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040

4141
doc_only_tf_supported = "(Supported Backend: TensorFlow) "
4242
doc_only_pt_supported = "(Supported Backend: PyTorch) "
43+
doc_only_pd_supported = "(Supported Backend: Paddle) "
4344
# descriptors
4445
doc_loc_frame = "Defines a local frame at each atom, and the compute the descriptor as local coordinates under this frame."
4546
doc_se_e2_a = "Used by the smooth edition of Deep Potential. The full relative coordinates are used to construct the descriptor."
@@ -3167,6 +3168,7 @@ def training_args(
31673168
doc_kf_blocksize = "The blocksize for the Kalman filter."
31683169
doc_model_prob = "The visiting probability of each model for each training step in the multi-task mode."
31693170
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."
31703172

31713173
arg_training_data = training_data_args()
31723174
arg_validation_data = validation_data_args()
@@ -3269,6 +3271,12 @@ def training_args(
32693271
optional=True,
32703272
doc=doc_only_pt_supported + doc_gradient_max_norm,
32713273
),
3274+
Argument(
3275+
"acc_freq",
3276+
int,
3277+
optional=True,
3278+
doc=doc_only_pd_supported + doc_acc_freq,
3279+
),
32723280
]
32733281
variants = [
32743282
Variant(

0 commit comments

Comments
 (0)