mirror of
https://github.com/facebookresearch/sam2.git
synced 2025-09-18 04:32:48 +08:00
Update hieradet.py
ufmt formatting fixed.
This commit is contained in:
parent
102ddb8899
commit
46945a2122
@ -47,7 +47,6 @@ 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
|
||||||
|
|
||||||
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)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user