[Fix] Fix a typo in the docstring of MSDeformAttn (open-mmlab#2599)

zhouzaida · Mar 20, 2023 · 7d3899b · 7d3899b
1 parent 27c87c2
commit 7d3899b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/multi_scale_deform_attn.py b/mmcv/ops/multi_scale_deform_attn.py
@@ -178,7 +178,7 @@ class MultiScaleDeformableAttention(BaseModule):
     Args:
         embed_dims (int): The embedding dimension of Attention.
             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
             Attention. Default: 4.
         num_points (int): The number of sampling points for