diff --git a/segment_anything/modeling/image_encoder.py b/segment_anything/modeling/image_encoder.py index 755ff4f..66351d9 100644 --- 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__()