mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Remove dead line

Browse Source
This commit is contained in:
Ross Wightman 2023-02-15 15:38:41 -08:00 committed by Ross Wightman
parent f77c04ff36
commit 2e38d53dca
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/models/vision_transformer.py
View File

@ -241,7 +241,6 @@ class ParallelScalingBlock(nn.Module):
self.fast_attn = hasattr(torch.nn.functional, 'scaled_dot_product_attention') # FIXME
mlp_hidden_dim = int(mlp_ratio * dim)
in_proj_out_dim = mlp_hidden_dim + 3 * dim
out_proj_in_dim = mlp_hidden_dim + dim
self.in_norm = norm_layer(dim)
self.in_proj = nn.Linear(dim, in_proj_out_dim, bias=qkv_bias)