update docs/models

brianhou0208 · brianhou0208 · commit 0bc0360e60e2 · 2024-10-12T21:27:52.000+08:00
diff --git a/docs/models.rst b/docs/models.rst
@@ -73,3 +73,11 @@ PAN
 UPerNet
 ~~~~~~~
 .. autoclass:: segmentation_models_pytorch.UPerNet
+
+
+.. _segformer:
+
+Segformer
+~~~~~~~~~
+.. autoclass:: segmentation_models_pytorch.Segformer
+
