Skip to content

Commit e850892

Browse files
johngrahamreynoldsJohn Graham Reynolds
andauthored
Update data collator docstrings to accurately reference Nvidia tensor core compute capability version (#35188)
update data collator docs to reflect correct tensor core compute capability Co-authored-by: John Graham Reynolds <john.graham.reynolds@vumc.org>
1 parent 5290f6a commit e850892

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

src/transformers/data/data_collator.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)