Corrected docstring typo of class PatchEmbed

2023-04-10 22:53:23 +09:30
parent 150e5812c7
commit c1910835a3
1 changed files with 1 additions and 1 deletions
--- a/segment_anything/modeling/image_encoder.py
+++ b/segment_anything/modeling/image_encoder.py
@@ -380,7 +380,7 @@ class PatchEmbed(nn.Module):
            stride (Tuple): stride of the projection layer.
            padding (Tuple): padding size of the projection layer.
            in_chans (int): Number of input image channels.
-            embed_dim (int):  embed_dim (int): Patch embedding dimension.
+            embed_dim (int): Patch embedding dimension.
        """
        super().__init__()