From 923565875e9a5757f7c6a55967edef231347d721 Mon Sep 17 00:00:00 2001 From: Lionel Date: Tue, 5 Nov 2024 17:25:29 +0100 Subject: [PATCH] make v1/v2 independent transforms available --- lightly/transforms/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lightly/transforms/__init__.py b/lightly/transforms/__init__.py index 6794c146f..81acff0fc 100644 --- a/lightly/transforms/__init__.py +++ b/lightly/transforms/__init__.py @@ -46,6 +46,7 @@ TiCoView1Transform, TiCoView2Transform, ) +from lightly.transforms.torchvision_transforms import torchvision_transforms from lightly.transforms.vicreg_transform import VICRegTransform, VICRegViewTransform from lightly.transforms.vicregl_transform import VICRegLTransform, VICRegLViewTransform from lightly.transforms.wmse_transform import WMSETransform