File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 4040
4141doc_only_tf_supported = "(Supported Backend: TensorFlow) "
4242doc_only_pt_supported = "(Supported Backend: PyTorch) "
43+ doc_only_pd_supported = "(Supported Backend: Paddle) "
4344# descriptors
4445doc_loc_frame = "Defines a local frame at each atom, and the compute the descriptor as local coordinates under this frame."
4546doc_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 (
You can’t perform that action at this time.
0 commit comments