transition from torchvision.transforms v1 to version indep

lightly-ai · Nov 6, 2024 · a7e8be2 · a7e8be2
1 parent 01fdac4
commit a7e8be2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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