From b82d8833c2a872acf1d73541032c57762dc5f0cc Mon Sep 17 00:00:00 2001 From: David Chiu Date: Thu, 24 Aug 2023 23:21:53 +0800 Subject: [PATCH] Fix typos in _augment.py (#7877) Co-authored-by: Nicolas Hug --- torchvision/transforms/v2/_augment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/transforms/v2/_augment.py b/torchvision/transforms/v2/_augment.py index 130950fee34..a9bad8f9bf7 100644 --- a/torchvision/transforms/v2/_augment.py +++ b/torchvision/transforms/v2/_augment.py @@ -229,7 +229,7 @@ class MixUp(_BaseMixUpCutMix): alpha (float, optional): hyperparameter of the Beta distribution used for mixup. Default is 1. num_classes (int): number of classes in the batch. Used for one-hot-encoding. labels_getter (callable or "default", optional): indicates how to identify the labels in the input. - By default, this will pick the second parameter a the labels if it's a tensor. This covers the most + By default, this will pick the second parameter as the labels if it's a tensor. This covers the most common scenario where this transform is called as ``MixUp()(imgs_batch, labels_batch)``. It can also be a callable that takes the same input as the transform, and returns the labels. """ @@ -279,7 +279,7 @@ class CutMix(_BaseMixUpCutMix): alpha (float, optional): hyperparameter of the Beta distribution used for mixup. Default is 1. num_classes (int): number of classes in the batch. Used for one-hot-encoding. labels_getter (callable or "default", optional): indicates how to identify the labels in the input. - By default, this will pick the second parameter a the labels if it's a tensor. This covers the most + By default, this will pick the second parameter as the labels if it's a tensor. This covers the most common scenario where this transform is called as ``CutMix()(imgs_batch, labels_batch)``. It can also be a callable that takes the same input as the transform, and returns the labels. """