@@ -256,7 +256,7 @@ class DataCollatorWithPadding:
256256 If set will pad the sequence to a multiple of the provided value.
257257
258258 This is especially useful to enable the use of Tensor Cores on NVIDIA hardware with compute capability >=
259- 7.5 (Volta).
259+ 7.0 (Volta).
260260 return_tensors (`str`, *optional*, defaults to `"pt"`):
261261 The type of Tensor to return. Allowable values are "np", "pt" and "tf".
262262 """
@@ -308,7 +308,7 @@ class DataCollatorForTokenClassification(DataCollatorMixin):
308308 If set will pad the sequence to a multiple of the provided value.
309309
310310 This is especially useful to enable the use of Tensor Cores on NVIDIA hardware with compute capability >=
311- 7.5 (Volta).
311+ 7.0 (Volta).
312312 label_pad_token_id (`int`, *optional*, defaults to -100):
313313 The id to use when padding the labels (-100 will be automatically ignore by PyTorch loss functions).
314314 return_tensors (`str`, *optional*, defaults to `"pt"`):
@@ -568,7 +568,7 @@ class DataCollatorForSeq2Seq:
568568 If set will pad the sequence to a multiple of the provided value.
569569
570570 This is especially useful to enable the use of Tensor Cores on NVIDIA hardware with compute capability >=
571- 7.5 (Volta).
571+ 7.0 (Volta).
572572 label_pad_token_id (`int`, *optional*, defaults to -100):
573573 The id to use when padding the labels (-100 will be automatically ignored by PyTorch loss functions).
574574 return_tensors (`str`, *optional*, defaults to `"pt"`):
@@ -693,6 +693,9 @@ class DataCollatorForLanguageModeling(DataCollatorMixin):
693693 The probability with which to (randomly) mask tokens in the input, when `mlm` is set to `True`.
694694 pad_to_multiple_of (`int`, *optional*):
695695 If set will pad the sequence to a multiple of the provided value.
696+
697+ This is especially useful to enable the use of Tensor Cores on NVIDIA hardware with compute capability >=
698+ 7.0 (Volta).
696699 return_tensors (`str`):
697700 The type of Tensor to return. Allowable values are "np", "pt" and "tf".
698701
0 commit comments