We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 63532c2 commit c7120f6Copy full SHA for c7120f6
timm/models/cvt.py
@@ -130,8 +130,8 @@ def __init__(
130
assert dim % num_heads == 0, 'dim should be divisible by num_heads'
131
self.num_heads = num_heads
132
self.head_dim = dim // num_heads
133
- self.scale = self.head_dim ** -0.5
134
- self.fused_attn = use_fused_attn()
+ self.scale = dim ** -0.5
+ self.fused_attn = False #use_fused_attn()
135
136
self.proj_q = nn.Linear(dim, dim, bias=qkv_bias)
137
self.proj_k = nn.Linear(dim, dim, bias=qkv_bias)
0 commit comments