File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ]
You can’t perform that action at this time.
0 commit comments