Fixed some typos in comments

2023-04-07 00:50:18 +08:00
parent aac76a1fb0
commit 20114f5138
8 changed files with 11 additions and 11 deletions
--- a/segment_anything/modeling/transformer.py
+++ b/segment_anything/modeling/transformer.py
@@ -96,7 +96,7 @@ class TwoWayTransformer(nn.Module):
                key_pe=image_pe,
            )

-        # Apply the final attenion layer from the points to the image
+        # Apply the final attention layer from the points to the image
        q = queries + point_embedding
        k = keys + image_pe
        attn_out = self.final_attn_token_to_image(q=q, k=k, v=keys)