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

pull/2615/head
Hakjin Lee 2023-02-13 17:59:25 +09:00 committed by GitHub
parent 5a61e534ea
commit cb94ffb672
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -167,7 +167,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