mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed a typo in multi_scale_deform_attn.py (#1838)

pull/1845/head
wulele 2022-03-28 15:35:50 +08:00 committed by GitHub
parent 25602c6844
commit c46deb0576
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcv/ops/multi_scale_deform_attn.py
View File

@ -95,7 +95,7 @@ def multi_scale_deformable_attn_pytorch(value, value_spatial_shapes,
Args:
value (torch.Tensor): The value has shape
(bs, num_keys, mum_heads, embed_dims//num_heads)
(bs, num_keys, num_heads, embed_dims//num_heads)
value_spatial_shapes (torch.Tensor): Spatial shape of
each feature map, has shape (num_levels, 2),
last dimension 2 represent (h, w)