Merge pull request #681 from lmk123568/master

Update convit.py
3 years ago · 45dec179e5
parent 4907f8f70d b87d98b238
commit 45dec179e5
1 changed files with 0 additions and 1 deletions
--- a/timm/models/convit.py
+++ b/timm/models/convit.py
@ -64,7 +64,6 @@ class GPSA(nn.Module):
        self.dim = dim
        head_dim = dim // num_heads
        self.scale = qk_scale or head_dim ** -0.5
-        self.locality_strength = locality_strength

        self.qk = nn.Linear(dim, dim * 2, bias=qkv_bias)
        self.v = nn.Linear(dim, dim, bias=qkv_bias)