From a7e8be2e63d2087a0c275fd52c50d118421fe072 Mon Sep 17 00:00:00 2001 From: Lionel Date: Wed, 6 Nov 2024 17:36:56 +0100 Subject: [PATCH] transition from torchvision.transforms v1 to version indep --- lightly/transforms/ijepa_transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightly/transforms/ijepa_transform.py b/lightly/transforms/ijepa_transform.py index 6bd002446..f1d28af81 100644 --- a/lightly/transforms/ijepa_transform.py +++ b/lightly/transforms/ijepa_transform.py @@ -1,9 +1,9 @@ from typing import Dict, List, Tuple, Union -import torchvision.transforms as T from PIL.Image import Image from torch import Tensor +from lightly.transforms.torchvision_v2_compatibility import torchvision_transforms as T from lightly.transforms.utils import IMAGENET_NORMALIZE