quoccuongle/Grounded-SAM-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #167 from arun477/patch-1

Browse Source 
remove unused attributes from hieradet.py
This commit is contained in:
Chay Ryali
2024-08-09 10:31:56 -07:00
committed by GitHub
parent d421e0b040 8f607e2de1
commit 778e112740
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sam2/modeling/backbones/hieradet.py
View File

@@ -46,11 +46,7 @@ class MultiScaleAttention(nn.Module):
self.dim = dim self.dim = dim
self.dim_out = dim_out self.dim_out = dim_out
self.num_heads = num_heads self.num_heads = num_heads
head_dim = dim_out // num_heads
self.scale = head_dim**-0.5
self.q_pool = q_pool self.q_pool = q_pool
self.qkv = nn.Linear(dim, dim_out * 3) self.qkv = nn.Linear(dim, dim_out * 3)
self.proj = nn.Linear(dim_out, dim_out) self.proj = nn.Linear(dim_out, dim_out)