diff --git a/docs/source/transforms.rst b/docs/source/transforms.rst index 0f6661c5198..7cda218c65a 100644 --- a/docs/source/transforms.rst +++ b/docs/source/transforms.rst @@ -95,7 +95,7 @@ For example, you can apply a functional transform to multiple images like this: import random def my_segmentation_transforms(image, segmentation): - if random.random() > 5: + if random.random() > 0.5: angle = random.randint(-30, 30) image = TF.rotate(image, angle) segmentation = TF.rotate(segmentation, angle)