Skip to content

Commit

Permalink
Merge pull request #7 from eltociear/patch-1
Browse files Browse the repository at this point in the history
chore: update upn_decoder.py
  • Loading branch information
Mountchicken authored Jan 24, 2025
2 parents b05d399 + 75e775c commit 049852b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions chatrex/upn/models/decoder/upn_decoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
class DeformableTransformerDecoderLayer(nn.Module):
"""Deformable Transformer Decoder Layer. This is a modified version in Grounding DINO.
After the query is attented to the image feature, it is further attented to the text feature.
The excute order is: self_attn -> cross_attn to text -> cross_attn to image -> ffn
The execute order is: self_attn -> cross_attn to text -> cross_attn to image -> ffn
Args:
d_model (int): The dimension of keys/values/queries in :class:`MultiheadAttention`.
d_ffn (int): The dimension of the feedforward network model.
Expand Down Expand Up @@ -145,7 +145,7 @@ def forward(
@DECODERS.register_module()
class UPNDecoder(nn.Module):
"""Decoder used in UPN. Each layer is a DeformableTransformerDecoderLayer. The query
will be abled to attend the image feature and text feature. The excute order is:
will be abled to attend the image feature and text feature. The execute order is:
self_attn -> cross_attn to image -> ffn
Args:
Expand Down

0 comments on commit 049852b

Please sign in to comment.