Update hieradet.py

ufmt formatting fixed.
This commit is contained in:
Arun
2024-08-09 11:14:11 +05:30
committed by GitHub
parent 102ddb8899
commit 46945a2122

View File

@@ -47,7 +47,6 @@ class MultiScaleAttention(nn.Module):
self.dim = dim
self.dim_out = dim_out
self.num_heads = num_heads
self.q_pool = q_pool
self.qkv = nn.Linear(dim, dim_out * 3)
self.proj = nn.Linear(dim_out, dim_out)