diff --git a/albumentations/augmentations/tk/transform.py b/albumentations/augmentations/tk/transform.py index 7a33b5476..0241debaa 100644 --- a/albumentations/augmentations/tk/transform.py +++ b/albumentations/augmentations/tk/transform.py @@ -1557,6 +1557,7 @@ class GaussianNoise(GaussNoise): References: - torchvision: https://pytorch.org/vision/master/generated/torchvision.transforms.v2.GaussianNoise.html + - kornia: https://kornia.readthedocs.io/en/latest/augmentation.module.html#kornia.augmentation.RandomGaussianNoise """ class InitSchema(BaseTransformInitSchema):