Show SSL-EY in docs

lightly-ai · Jan 5, 2024 · f88efd0 · f88efd0
1 parent e8e3054
commit f88efd0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/docs/source/examples/models.rst b/docs/source/examples/models.rst
@@ -24,6 +24,7 @@ for PyTorch and PyTorch Lightning to give you a headstart when implementing your
     simmim.rst
     simsiam.rst
     smog.rst
+    ssley.rst
     swav.rst
     tico.rst
     vicreg.rst

diff --git a/docs/source/lightly.transforms.rst b/docs/source/lightly.transforms.rst
@@ -71,6 +71,10 @@ lightly.transforms
    :members:
    :special-members: __call__
 
+.. automodule:: lightly.transforms.ssley_transform
+   :members:
+   :special-members: __call__
+
 .. automodule:: lightly.transforms.swav_transform
    :members:
    :special-members: __call__