mirror of https://github.com/open-mmlab/mmcv.git
[Fix] Fix a typo in the docstring of MSDeformAttn (#2599)
parent
5a61e534ea
commit
cb94ffb672
|
@ -167,7 +167,7 @@ class MultiScaleDeformableAttention(BaseModule):
|
||||||
Args:
|
Args:
|
||||||
embed_dims (int): The embedding dimension of Attention.
|
embed_dims (int): The embedding dimension of Attention.
|
||||||
Default: 256.
|
Default: 256.
|
||||||
num_heads (int): Parallel attention heads. Default: 64.
|
num_heads (int): Parallel attention heads. Default: 8.
|
||||||
num_levels (int): The number of feature map used in
|
num_levels (int): The number of feature map used in
|
||||||
Attention. Default: 4.
|
Attention. Default: 4.
|
||||||
num_points (int): The number of sampling points for
|
num_points (int): The number of sampling points for
|
||||||
|
|
Loading…
Reference in New Issue