From 2e0d6ce25408efe89c17eaeac0997cac92446bb5 Mon Sep 17 00:00:00 2001 From: "Kai (Kazuya Ito)" Date: Fri, 7 Nov 2025 16:51:27 +0900 Subject: [PATCH] Update _auto_augment.py --- torchvision/transforms/v2/_auto_augment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/transforms/v2/_auto_augment.py b/torchvision/transforms/v2/_auto_augment.py index 52707af1f2e..f2a9a31bfd1 100644 --- a/torchvision/transforms/v2/_auto_augment.py +++ b/torchvision/transforms/v2/_auto_augment.py @@ -517,7 +517,7 @@ class AugMix(_AutoAugmentBase): Args: severity (int, optional): The severity of base augmentation operators. Default is ``3``. mixture_width (int, optional): The number of augmentation chains. Default is ``3``. - chain_depth (int, optional): The depth of augmentation chains. A negative value denotes stochastic depth sampled from the interval [1, 3]. + chain_depth (int, optional): The depth of augmentation chains. A zero or negative value denotes stochastic depth sampled from the interval [1, 3]. Default is ``-1``. alpha (float, optional): The hyperparameter for the probability distributions. Default is ``1.0``. all_ops (bool, optional): Use all operations (including brightness, contrast, color and sharpness). Default is ``True``.