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