proj_drop wasn't being called in sam vit

huggingface · Aug 2, 2024 · 1034462 · 1034462
1 parent a6fe31b
commit 1034462
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/timm/models/vision_transformer_sam.py b/timm/models/vision_transformer_sam.py
@@ -182,6 +182,7 @@ def forward(self, x):
 
         x = x.view(B, self.num_heads, N, -1).transpose(1, 2).reshape(B, N, -1)
         x = self.proj(x)
+        x = self.proj_drop(x)
         x = x.view(B, H, W, -1)
         return x